[text] xl

Viewer

  1. import pandas as pd
  2. from langchain.agents import create_pandas_dataframe_agent
  3. from langchain.chat_models import ChatOpenAI
  4. from langchain.llms import OpenAI
  5. from langchain.agents import create_csv_agent
  6. from langchain.agents.agent_types import AgentType
  7.  
  8. import openai
  9. import tiktoken
  10. import os
  11.  
  12. os.environ["OPENAI_API_KEY"] = "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"
  13. openai.api_key="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"
  14.  
  15.  
  16. # csv_agent = create_csv_agent(ChatOpenAI(temperature=0), 'Page2.csv', verbose=True,return_intermediate_steps=True,  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION)
  17. # response = csv_agent({"input":"Give me Total net revenue for 1Q23,4Q22,3Q22"})
  18.  
  19. # print(response["output"])
  20.  
  21. model_id = "gpt-3.5-turbo"
  22. df = pd.read_csv('Page2.csv', encoding= 'unicode_escape')
  23. # print(df.columns)
  24.  
  25. llm = OpenAI(temperature=0,model_name="gpt-3.5-turbo")
  26.  
  27. agent = create_pandas_dataframe_agent(
  28.     llm,
  29.     df,
  30.     verbose=True
  31. )
  32. agent.run("From selected income statement row could you provide Total net revenue for 1Q23,4Q22,3Q22")
  33.  

Editor

You can edit this paste and save as new:


File Description
  • xl
  • Paste Code
  • 09 May-2024
  • 1.07 Kb
You can Share it: