From f3ac6493e815462728aa14bf307422c1e8de915d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 19 Jul 2024 16:18:53 -0700 Subject: [PATCH] fix typing errors --- litellm/batches/main.py | 2 +- litellm/types/llms/openai.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/batches/main.py b/litellm/batches/main.py index af2dc5059d3..db220e15faa 100644 --- a/litellm/batches/main.py +++ b/litellm/batches/main.py @@ -89,7 +89,7 @@ async def acreate_batch( def create_batch( completion_window: Literal["24h"], - endpoint: Literal["/v1/chat/completions", "/v1/embeddings"], + endpoint: Literal["/v1/chat/completions", "/v1/embeddings", "/v1/completions"], input_file_id: str, custom_llm_provider: Literal["openai"] = "openai", metadata: Optional[Dict[str, str]] = None, diff --git a/litellm/types/llms/openai.py b/litellm/types/llms/openai.py index 42f1dac3d6d..294e299dbf4 100644 --- a/litellm/types/llms/openai.py +++ b/litellm/types/llms/openai.py @@ -257,7 +257,7 @@ class CreateBatchRequest(TypedDict, total=False): """ completion_window: Literal["24h"] - endpoint: Literal["/v1/chat/completions", "/v1/embeddings"] + endpoint: Literal["/v1/chat/completions", "/v1/embeddings", "/v1/completions"] input_file_id: str metadata: Optional[Dict[str, str]] extra_headers: Optional[Dict[str, str]]