ha1772007 commited on
Commit
1d582be
·
verified ·
1 Parent(s): f7b213b

Update chat.py

Browse files
Files changed (1) hide show
  1. chat.py +1 -1
chat.py CHANGED
@@ -13,6 +13,6 @@ def langchainConversation(conversation):
13
  def converse(conversation,provider,model,key):
14
  if(provider=='groq'):
15
  chat = ChatGroq(temperature=0, groq_api_key=key, model_name=model)
16
- elif(provider=='google'):
17
  chat = ChatGoogleGenerativeAI(model=model,google_api_key=key)
18
  return json.dumps(json.loads(chat.invoke(langchainConversation(conversation)).json()),indent=4)
 
13
  def converse(conversation,provider,model,key):
14
  if(provider=='groq'):
15
  chat = ChatGroq(temperature=0, groq_api_key=key, model_name=model)
16
+ elif(provider=='gemini'):
17
  chat = ChatGoogleGenerativeAI(model=model,google_api_key=key)
18
  return json.dumps(json.loads(chat.invoke(langchainConversation(conversation)).json()),indent=4)