fadliaulawi
commited on
Commit
•
54d5e6e
1
Parent(s):
ca71749
Fix check llm_val
Browse files- process.py +1 -1
process.py
CHANGED
@@ -42,7 +42,7 @@ class Process():
|
|
42 |
|
43 |
if llm_val.startswith('gpt'):
|
44 |
self.llm_val = ChatOpenAI(temperature=0, model_name=llm_val)
|
45 |
-
elif
|
46 |
self.llm_val = ChatGoogleGenerativeAI(temperature=0, model=llm_val)
|
47 |
else:
|
48 |
self.llm_val = ChatOpenAI(temperature=0, model_name=llm_val, api_key=os.environ['PERPLEXITY_API_KEY'], base_url="https://api.perplexity.ai")
|
|
|
42 |
|
43 |
if llm_val.startswith('gpt'):
|
44 |
self.llm_val = ChatOpenAI(temperature=0, model_name=llm_val)
|
45 |
+
elif llm_val.startswith('gemini'):
|
46 |
self.llm_val = ChatGoogleGenerativeAI(temperature=0, model=llm_val)
|
47 |
else:
|
48 |
self.llm_val = ChatOpenAI(temperature=0, model_name=llm_val, api_key=os.environ['PERPLEXITY_API_KEY'], base_url="https://api.perplexity.ai")
|