diff --git a/app/llm/wrapper/__init__.py b/app/llm/wrapper/__init__.py index cd3c3f61..23852852 100644 --- a/app/llm/wrapper/__init__.py +++ b/app/llm/wrapper/__init__.py @@ -11,11 +11,11 @@ from llm.wrapper.ollama import OllamaModel type AnyLLM = ( - DirectOpenAICompletionModel - | AzureOpenAICompletionModel - | DirectOpenAIChatModel - | AzureOpenAIChatModel - | DirectOpenAIEmbeddingModel - | AzureOpenAIEmbeddingModel - | OllamaModel + DirectOpenAICompletionModel + | AzureOpenAICompletionModel + | DirectOpenAIChatModel + | AzureOpenAIChatModel + | DirectOpenAIEmbeddingModel + | AzureOpenAIEmbeddingModel + | OllamaModel )