datacipen commited on
Commit
6037cc1
·
verified ·
1 Parent(s): 469ec92

Delete csvanswer.py

Browse files
Files changed (1) hide show
  1. csvanswer.py +0 -59
csvanswer.py DELETED
@@ -1,59 +0,0 @@
1
- import os
2
- import pandas as pd
3
-
4
- from langchain_openai import ChatOpenAI
5
-
6
- from langchain.agents import AgentExecutor
7
- from langchain.agents.agent_types import AgentType
8
- from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent, create_csv_agent
9
-
10
- import chainlit as cl
11
-
12
- from deep_translator import GoogleTranslator
13
-
14
- async def create_agent(filename: str):
15
- """
16
- Create an agent that can access and use a large language model (LLM).
17
-
18
- Args:
19
- filename: The path to the CSV file that contains the data.
20
-
21
- Returns:
22
- An agent that can access and use the LLM.
23
- """
24
-
25
- # Create an OpenAI object.
26
- os.environ['OPENAI_API_KEY'] = os.environ['OPENAI_API_KEY']
27
- llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
28
-
29
- # Read the CSV file into a Pandas DataFrame.
30
- df = pd.read_csv(filename)
31
-
32
- # Create a Pandas DataFrame agent.
33
- return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
34
-
35
-
36
- async def LLMAnswer(msg):
37
- agent = await create_agent("./public/surveyia.csv")
38
- cb = cl.AsyncLangchainCallbackHandler()
39
- try:
40
- res = await agent.acall("Réponds en langue française à la question suivante : " + msg, callbacks=[cb])
41
- await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(res['output'])).send()
42
- except ValueError as e:
43
- res = str(e)
44
- resArray = res.split(":")
45
- ans = ''
46
- if str(res).find('parsing') != -1:
47
- for i in range(2,len(resArray)):
48
- ans += resArray[i]
49
- await cl.Message(author="COPILOT",content=ans.replace("`","")).send()
50
- else:
51
- await cl.Message(author="COPILOT",content="Reformulez votre requête, s'il vous plait 😃").send()
52
- # Query the agent.
53
- #response = query_agent(agent=agent, query=message.content)
54
- # Decode the response.
55
- #decoded_response = decode_response(response)
56
-
57
- # Write the response to the Streamlit app.
58
- #result = write_response(decoded_response)
59
- #await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(result)).send()