Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -49,10 +49,7 @@ class BasicAgent:
|
|
49 |
|
50 |
# LLM per LlamaIndex
|
51 |
llm = OpenAI(
|
52 |
-
|
53 |
-
temperature=0,
|
54 |
-
api_key=openai_api_key,
|
55 |
-
verbose=True
|
56 |
)
|
57 |
Settings.llm = llm
|
58 |
|
@@ -218,7 +215,11 @@ class BasicAgent:
|
|
218 |
|
219 |
def _ask_gpt4o(self, text: str) -> str:
|
220 |
messages = [{"role": "user", "content": text}]
|
221 |
-
response = self.client.chat.completions.create(
|
|
|
|
|
|
|
|
|
222 |
return response.choices[0].message.content.strip()
|
223 |
|
224 |
def _ask_gpt4o_with_image(self, image: Image.Image, question: str) -> str:
|
@@ -229,6 +230,7 @@ class BasicAgent:
|
|
229 |
|
230 |
response = self.client.chat.completions.create(
|
231 |
model="gpt-4o", #ATTENZIONE QUI MODELLO NON MINI
|
|
|
232 |
messages=[{
|
233 |
"role": "user",
|
234 |
"content": [
|
|
|
49 |
|
50 |
# LLM per LlamaIndex
|
51 |
llm = OpenAI(
|
52 |
+
api_key=openai_api_key
|
|
|
|
|
|
|
53 |
)
|
54 |
Settings.llm = llm
|
55 |
|
|
|
215 |
|
216 |
def _ask_gpt4o(self, text: str) -> str:
|
217 |
messages = [{"role": "user", "content": text}]
|
218 |
+
response = self.client.chat.completions.create(
|
219 |
+
model="gpt-4o-mini",
|
220 |
+
temperature=0,
|
221 |
+
messages=messages
|
222 |
+
)
|
223 |
return response.choices[0].message.content.strip()
|
224 |
|
225 |
def _ask_gpt4o_with_image(self, image: Image.Image, question: str) -> str:
|
|
|
230 |
|
231 |
response = self.client.chat.completions.create(
|
232 |
model="gpt-4o", #ATTENZIONE QUI MODELLO NON MINI
|
233 |
+
temperature=0,
|
234 |
messages=[{
|
235 |
"role": "user",
|
236 |
"content": [
|