Merge pull request #24 from ahuang11/patch-2

Fix pop model_name
shroominic · Feb 11, 2024 · 9ade305 · 9ade305
2 parents f7effee + 860f885
commit 9ade305
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/funcchain/model/defaults.py b/src/funcchain/model/defaults.py
@@ -155,7 +155,7 @@ def univeral_model_selector(
         except Exception as e:
             print(e)
 
-    model_kwargs.pop("model_name")
+    model_kwargs.pop("model_name", None)
 
     if settings.openai_api_key:
         from langchain_openai.chat_models import ChatOpenAI