crazyforprogramming commited on
Commit
051072d
·
verified ·
1 Parent(s): af21af8

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -7,6 +7,7 @@ import json
7
  import requests
8
  from pypdf import PdfReader
9
  import gradio as gr
 
10
 
11
 
12
  load_dotenv(override=True)
@@ -146,10 +147,13 @@ If the user is engaging in discussion, try to steer them towards getting in touc
146
  return system_prompt
147
 
148
  def chat(self, message, history):
 
 
149
  messages = [{"role": "system", "content": self.system_prompt()}] + history + [{"role": "user", "content": message}]
150
  done = False
151
  while not done:
152
- response = self.openai.chat.completions.create(model="gpt-4o-mini", messages=messages, tools=tools)
 
153
  if response.choices[0].finish_reason=="tool_calls":
154
  message = response.choices[0].message
155
  tool_calls = message.tool_calls
 
7
  import requests
8
  from pypdf import PdfReader
9
  import gradio as gr
10
+ import os
11
 
12
 
13
  load_dotenv(override=True)
 
147
  return system_prompt
148
 
149
  def chat(self, message, history):
150
+ google_api_key = os.getenv('GOOGLE_API_KEY')
151
+ gemini = OpenAI(api_key=google_api_key, base_url="https://generativelanguage.googleapis.com/v1beta/openai/")
152
  messages = [{"role": "system", "content": self.system_prompt()}] + history + [{"role": "user", "content": message}]
153
  done = False
154
  while not done:
155
+ # response = self.openai.chat.completions.create(model="gpt-4o-mini", messages=messages, tools=tools)
156
+ response = gemini.chat.completions.create(model="gemini-2.5-flash", messages=messages, tools=tools)
157
  if response.choices[0].finish_reason=="tool_calls":
158
  message = response.choices[0].message
159
  tool_calls = message.tool_calls