diff --git a/litellm/llms/ai21.py b/litellm/llms/ai21.py index 72708b832..22f073dfc 100644 --- a/litellm/llms/ai21.py +++ b/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: diff --git a/litellm/llms/aleph_alpha.py b/litellm/llms/aleph_alpha.py index 090262461..253a17613 100644 --- a/litellm/llms/aleph_alpha.py +++ b/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 diff --git a/litellm/llms/ollama.py b/litellm/llms/ollama.py index c46a343b4..e4639d073 100644 --- a/litellm/llms/ollama.py +++ b/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