GattoNero commited on
Commit
c044f85
·
verified ·
1 Parent(s): d0c9345

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -4,6 +4,8 @@ import requests
4
  import inspect
5
  import pandas as pd
6
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, OpenAIServerModel
 
 
7
 
8
 
9
  # (Keep Constants as is)
@@ -18,7 +20,11 @@ class BasicAgent:
18
  print("Initializing LlamaIndex-based agent...")
19
 
20
  # Imposta l'LLM (puoi usare anche altri modelli via HuggingFace o OpenRouter)
21
- self.llm = HfApiModel()
 
 
 
 
22
 
23
  #OpenAI(model="gpt-3.5-turbo", temperature=0)
24
 
 
4
  import inspect
5
  import pandas as pd
6
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, OpenAIServerModel
7
+ from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
8
+ from llama_index.llms.openai import OpenAI
9
 
10
 
11
  # (Keep Constants as is)
 
20
  print("Initializing LlamaIndex-based agent...")
21
 
22
  # Imposta l'LLM (puoi usare anche altri modelli via HuggingFace o OpenRouter)
23
+ self.llm = OpenAI(
24
+ model="gpt-4o-mini",
25
+ temperature=0,
26
+ api_key=openai_api_key
27
+ )
28
 
29
  #OpenAI(model="gpt-3.5-turbo", temperature=0)
30