serverdaun commited on
Commit
0e29489
·
1 Parent(s): 5bf7ac6

Add configuration file to load environment variables for OpenAI models and memory file

Browse files
Files changed (2) hide show
  1. config.py +9 -0
  2. tools.py +0 -4
config.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ from dotenv import load_dotenv
4
+
5
+ load_dotenv(override=True)
6
+
7
+ OPENAI_CHAT_MODEL_WORKER = os.getenv("OPENAI_CHAT_MODEL_WORKER", "gpt-4o-mini")
8
+ OPENAI_CHAT_MODEL_EVALUATOR = os.getenv("OPENAI_CHAT_MODEL_EVALUATOR", "gpt-4o-mini")
9
+ MEMORY_FILE = os.getenv("MEMORY_FILE", "sidekick_memory.json")
tools.py CHANGED
@@ -1,7 +1,6 @@
1
  import os
2
  from typing import List, Optional, Tuple
3
 
4
- from dotenv import load_dotenv
5
  from langchain.agents import Tool
6
  from langchain_community.agent_toolkits import (
7
  FileManagementToolkit,
@@ -15,9 +14,6 @@ from langchain_experimental.tools import PythonREPLTool
15
  from playwright.async_api import async_playwright
16
  from sympy import SympifyError, sympify
17
 
18
- # Load environment variables
19
- load_dotenv(override=True)
20
-
21
 
22
  # Custom math tool
23
  def safe_math_calculator(expression: str) -> str:
 
1
  import os
2
  from typing import List, Optional, Tuple
3
 
 
4
  from langchain.agents import Tool
5
  from langchain_community.agent_toolkits import (
6
  FileManagementToolkit,
 
14
  from playwright.async_api import async_playwright
15
  from sympy import SympifyError, sympify
16
 
 
 
 
17
 
18
  # Custom math tool
19
  def safe_math_calculator(expression: str) -> str: