update typing hint

HUBioDataLab · Feb 16, 2025 · b06b080 · b06b080
1 parent 29deb2c
commit b06b080
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/crossbar_llm/backend/tools/langchain_llm_qa_trial.py b/crossbar_llm/backend/tools/langchain_llm_qa_trial.py
@@ -257,6 +257,7 @@ def __init__(
             ReplicateLanguageModel,
             OllamaLanguageModel,
             NVIDIALanguageModel,
+            OpenRouterLanguageModel,
         ],
         qa_llm: Union[
             OpenAILanguageModel,
@@ -266,6 +267,7 @@ def __init__(
             ReplicateLanguageModel,
             OllamaLanguageModel,
             NVIDIALanguageModel,
+            OpenRouterLanguageModel,
         ],
         schema: dict,
         verbose: bool = False,