Ankush K commited on
Commit
55f3fb1
·
1 Parent(s): 9d923ce

move model details to env variable

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -76,9 +76,10 @@ tools = [{"type": "function", "function": record_user_details_json},
76
  class Me:
77
 
78
  def __init__(self):
79
- deepseek_api_key = os.getenv('DEEPSEEK_API_KEY')
80
- self.model_client = OpenAI(api_key=deepseek_api_key, base_url="https://api.deepseek.com/v1")
81
- self.model_name = "deepseek-chat"
 
82
  self.name = "Ankush Khobragade"
83
  with open("me/summary.txt", "r", encoding="utf-8") as f:
84
  self.summary = f.read()
 
76
  class Me:
77
 
78
  def __init__(self):
79
+ model_api_key = os.getenv('MODEL_API_KEY')
80
+ model_base_url = os.getenv('MODEL_BASE_URL')
81
+ self.model_client = OpenAI(api_key=model_api_key, base_url=model_base_url)
82
+ self.model_name = os.getenv('MODEL_NAME')
83
  self.name = "Ankush Khobragade"
84
  with open("me/summary.txt", "r", encoding="utf-8") as f:
85
  self.summary = f.read()