rrg92 commited on
Commit
db0dc73
·
verified ·
1 Parent(s): c40b095

Parametrize model

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -19,10 +19,9 @@ SqlDatabase = os.environ['SQL_DB']
19
  SqlUser = os.environ['SQL_USER']
20
  SqlPass = os.environ['SQL_PASS']
21
 
22
-
23
  OpenaiApiKey = os.environ.get("OPENAI_API_KEY")
24
  OpenaiBaseUrl = os.environ.get("OPENAI_BASE_URL","https://generativelanguage.googleapis.com/v1beta/openai")
25
-
26
 
27
  def sql(query,db=SqlDatabase, login_timeout = 120,onConnectionError = None):
28
 
@@ -76,7 +75,7 @@ def llm(messages, ResponseFormat = None, **kwargs):
76
  fn = ClientOpenai.beta.chat.completions.parse
77
 
78
  params = {
79
- 'model':"gemini-2.0-flash"
80
  ,'n':1
81
  ,'messages':messages
82
  ,'response_format':ResponseFormat
 
19
  SqlUser = os.environ['SQL_USER']
20
  SqlPass = os.environ['SQL_PASS']
21
 
 
22
  OpenaiApiKey = os.environ.get("OPENAI_API_KEY")
23
  OpenaiBaseUrl = os.environ.get("OPENAI_BASE_URL","https://generativelanguage.googleapis.com/v1beta/openai")
24
+ OpenaiDefaultModel = os.environ.get("OPENAI_MODEL","gemini-2.0-flash")
25
 
26
  def sql(query,db=SqlDatabase, login_timeout = 120,onConnectionError = None):
27
 
 
75
  fn = ClientOpenai.beta.chat.completions.parse
76
 
77
  params = {
78
+ 'model': OpenaiDefaultModel
79
  ,'n':1
80
  ,'messages':messages
81
  ,'response_format':ResponseFormat