Soumik555 commited on
Commit
4139a7c
·
1 Parent(s): d276167

added csv-agent-default in next.js

Browse files
Files changed (1) hide show
  1. gemini_report_generator.py +1 -1
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, conversation_history=conversation_history)
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"}