added csv-agent-default in next.js
Browse files
gemini_report_generator.py
CHANGED
@@ -270,7 +270,7 @@ class RethinkAgent(BaseModel):
|
|
270 |
def gemini_llm_chat(csv_url: str, query: str, conversation_history: List[Dict[str, Any]]) -> Dict[str, Any]:
|
271 |
try:
|
272 |
df = pd.read_csv(csv_url)
|
273 |
-
agent = RethinkAgent(df=df,
|
274 |
|
275 |
if not agent.initialize_model(API_KEYS):
|
276 |
return {"error": "API configuration failed"}
|
|
|
270 |
def gemini_llm_chat(csv_url: str, query: str, conversation_history: List[Dict[str, Any]]) -> Dict[str, Any]:
|
271 |
try:
|
272 |
df = pd.read_csv(csv_url)
|
273 |
+
agent = RethinkAgent(df=df, conversation=conversation_history)
|
274 |
|
275 |
if not agent.initialize_model(API_KEYS):
|
276 |
return {"error": "API configuration failed"}
|