GattoNero commited on
Commit
a344527
·
verified ·
1 Parent(s): ce9fd78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -68,10 +68,10 @@ class BasicAgent:
68
  self.agent = OpenAIAgent.from_tools([ingredient_tool], llm=llm, verbose=True)
69
 
70
  # Client OpenAI per chiamate esterne (immagini/audio)
71
- print_coso("prima di client")
72
  self.client = OpenAIClient(api_key=openai_api_key) # per .chat, .audio, ecc.
73
- print_coso("DOPO di client")
74
- Settings.llm = llm #LlamaOpenAI(model="gpt-4o-mini", api_key=openai_api_key) # per llamaindex
75
 
76
  # Carica i documenti
77
  self.documents = SimpleDirectoryReader("data").load_data()
@@ -223,6 +223,8 @@ class BasicAgent:
223
 
224
 
225
  def _ask_gpt4o(self, text: str) -> str:
 
 
226
  messages = [{"role": "user", "content": text}]
227
  response = self.client.chat.completions.create(
228
  model="gpt-4o-mini",
@@ -230,6 +232,7 @@ class BasicAgent:
230
  messages=messages
231
  )
232
  return response.choices[0].message.content.strip()
 
233
 
234
  def _ask_gpt4o_with_image(self, image: Image.Image, question: str) -> str:
235
  buffered = BytesIO()
 
68
  self.agent = OpenAIAgent.from_tools([ingredient_tool], llm=llm, verbose=True)
69
 
70
  # Client OpenAI per chiamate esterne (immagini/audio)
71
+
72
  self.client = OpenAIClient(api_key=openai_api_key) # per .chat, .audio, ecc.
73
+
74
+ Settings.llm = llm
75
 
76
  # Carica i documenti
77
  self.documents = SimpleDirectoryReader("data").load_data()
 
223
 
224
 
225
  def _ask_gpt4o(self, text: str) -> str:
226
+ self.agent.chat(text)
227
+ '''
228
  messages = [{"role": "user", "content": text}]
229
  response = self.client.chat.completions.create(
230
  model="gpt-4o-mini",
 
232
  messages=messages
233
  )
234
  return response.choices[0].message.content.strip()
235
+ '''
236
 
237
  def _ask_gpt4o_with_image(self, image: Image.Image, question: str) -> str:
238
  buffered = BytesIO()