refactor(ai21,-aleph-alpha,-ollama): making ai21, aleph-alpha, ollama compatible with openai v1 sdk

litellm/llms/ai21.py

@@ -11,7 +11,7 @@ class AI21Error(Exception):
     def __init__(self, status_code, message):
         self.status_code = status_code
         self.message = message
-        self.request = httpx.Request(method="POST", url="https://api.replicate.com/v1/deployments")
+        self.request = httpx.Request(method="POST", url="https://api.ai21.com/studio/v1/")
         self.response = httpx.Response(status_code=status_code, request=self.request)
         super().__init__(
             self.message
@@ -152,7 +152,6 @@ def completion(
                 original_response=response.text,
                 additional_args={"complete_input_dict": data},
             )
-        print_verbose(f"raw model_response: {response.text}")
         ## RESPONSE OBJECT
         completion_response = response.json()
         if "error" in completion_response:

litellm/llms/aleph_alpha.py

@@ -6,11 +6,14 @@ import time
 from typing import Callable, Optional
 import litellm
 from litellm.utils import ModelResponse, Choices, Message
+import httpx
 
 class AlephAlphaError(Exception):
     def __init__(self, status_code, message):
         self.status_code = status_code
         self.message = message
+        self.request = httpx.Request(method="POST", url="https://api.aleph-alpha.com/complete")
+        self.response = httpx.Response(status_code=status_code, request=self.request)
         super().__init__(
             self.message
         )  # Call the base class constructor with the parameters it needs

litellm/llms/ollama.py

@@ -3,6 +3,7 @@ import json
 import traceback
 from typing import Optional
 import litellm 
+import httpx
 
 try:
     from async_generator import async_generator, yield_  # optional dependency
@@ -14,6 +15,8 @@ class OllamaError(Exception):
     def __init__(self, status_code, message):
         self.status_code = status_code
         self.message = message
+        self.request = httpx.Request(method="POST", url="http://localhost:11434")
+        self.response = httpx.Response(status_code=status_code, request=self.request)
         super().__init__(
             self.message
         )  # Call the base class constructor with the parameters it needs