Change model_name to use environment variable

coolbeevip · Jul 10, 2024 · 90069e9 · 90069e9
1 parent 8d8436a
commit 90069e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py b/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py
@@ -14,7 +14,7 @@ class TestMarketingAnalysisAssistant(TestCase):
     def test_main(self):
         assistant = MarketingAnalysisAssistant(openai_api_base=os.environ["OPENAI_API_BASE"],
                                                openai_api_key=os.environ["OPENAI_API_KEY"],
-                                               model_name="gpt-4o",
+                                               model_name=os.environ["MODEL_NAME"],
                                                recursion_limit=50)
         assistant.run()