Upload folder using huggingface_hub
Browse files
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
|