feat(config): add support for NVIDIA and OpenRouter API keys

HUBioDataLab · Feb 17, 2025 · 7864872 · 7864872
1 parent b06b080
commit 7864872
Showing 1 changed file with 4 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
@@ -605,6 +605,8 @@ def run_for_query(
             self.config.anthropic_api_key = api_key
             self.config.groq_api_key = api_key
             self.config.replicate_api_key = api_key
+            self.config.nvidia_api_key = api_key
+            self.config.openrouter_api_key = api_key
 
         else:
             self.config = Config()
@@ -662,6 +664,8 @@ def execute_query(
             self.config.anthropic_api_key = api_key
             self.config.groq_api_key = api_key
             self.config.replicate_api_key = api_key
+            self.config.nvidia_api_key = api_key
+            self.config.openrouter_api_key = api_key
         else:
             self.config = Config()