fadliaulawi commited on
Commit
54d5e6e
1 Parent(s): ca71749

Fix check llm_val

Browse files
Files changed (1) hide show
  1. 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 llm.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")
 
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")