GattoNero commited on
Commit
09826d1
·
verified ·
1 Parent(s): d4b5ba8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -49,10 +49,7 @@ class BasicAgent:
49
 
50
  # LLM per LlamaIndex
51
  llm = OpenAI(
52
- model="gpt-4o-mini",
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(model="gpt-4o-mini", messages=messages)
 
 
 
 
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": [