Spaces:
Running
on
Zero
Running
on
Zero
Parametrize model
Browse files
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':
|
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
|