Merge pull request #404 from LlmKira/dev

♻️ refactor: add global credential tool model in credential.py…
LlmKira · Apr 20, 2024 · 4beda09 · 4beda09
2 parents b049286 + 9052dc5
commit 4beda09
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.env.exp b/.env.exp
@@ -1,6 +1,7 @@
 # NOTE:If you want to share your bot to **everyone**
 # GLOBAL_OAI_KEY=sk-xxx
-# GLOBAL_OAI_MODEL=gpt-3.5-turbo-0613
+GLOBAL_OAI_MODEL=gpt-3.5-turbo
+GLOBAL_OAI_TOOL_MODEL=gpt-3.5-turbo
 # GLOBAL_OAI_ENDPOINT=https://api.openai.com/v1/
 
 AMQP_DSN=amqp://admin:8a8a8a@localhost:5672/

diff --git a/app/components/credential.py b/app/components/credential.py
@@ -48,6 +48,7 @@ def from_provider(cls, token, provider_url):
         api_key=os.getenv("GLOBAL_OAI_KEY"),
         api_endpoint=os.getenv("GLOBAL_OAI_ENDPOINT"),
         api_model=os.getenv("GLOBAL_OAI_MODEL", "gpt-3.5-turbo"),
+        api_tool_model=os.getenv("GLOBAL_OAI_TOOL_MODEL", "gpt-3.5-turbo"),
     )
 else:
     global_credential = None