uyen13 commited on
Commit
0abbf38
·
verified ·
1 Parent(s): 519ae16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -20
app.py CHANGED
@@ -1,21 +1,25 @@
1
- from langchain import hub
2
- from langchain.agents import AgentExecutor, create_json_chat_agent
3
- from langchain_community.tools.tavily_search import TavilySearchResults
4
- from langchain_openai import ChatOpenAI
 
5
 
6
- # Define your API key for OpenAI or any other LLM provider
7
- OPENAI_API_KEY = 'sk-nAqoChT9cmkAxALwMLdWT3BIbkFJcNHsH5Z5LN2ixPcDAopT'
8
- openai.api_key=OPENAI_API_KEY
9
-
10
- tools = [TavilySearchResults(max_results=1)]
11
-
12
- # Choose the LLM that will drive the agent
13
- llm = ChatOpenAI(openai_api_key=openai.api_key,temperature=0)
14
-
15
- # Construct the JSON agent
16
- agent = create_json_chat_agent(llm, tools, prompt)
17
- # Create an agent executor by passing in the agent and tools
18
- agent_executor = AgentExecutor(
19
- agent=agent, tools=tools, verbose=True, handle_parsing_errors=True
20
- )
21
- agent_executor.invoke({"input": "what is LangChain?"})
 
 
 
 
1
+ from langchain.chat_models import ChatOpenAI
2
+ from langchain.agents import create_json_agent
3
+ from langchain.agents.agent_toolkits import JsonToolkit
4
+ from langchain.tools.json.tool import JsonSpec
5
+ import json
6
 
7
+ # file="output.json"
8
+ # with open(file,"r") as f1:
9
+ # data=json.load(f1)
10
+ # f1.close()
11
+ data ={
12
+ "openapi": "3.0.0",
13
+ "info": {
14
+ "title": "OpenAI API",
15
+ "description": "The OpenAI REST API. Please see https://platform.openai.com/docs/api-reference for more details.",
16
+ "version": "2.0.0",
17
+ "termsOfService": "https://openai.com/policies/terms-of-use",
18
+ "contact": {
19
+ "name": "OpenAI Support",
20
+ "url": "https://help.openai.com/"
21
+ }}}
22
+ spec=JsonSpec(dict_=data,max_value_length=4000)
23
+ toolkit=JsonToolkit(spec=spec)
24
+ agent=create_json_agent(llm=ChatOpenAI(openai_api_key="sk-nAqoChT9cmkAxALwMLdWT3BIbkFJcNHsH5Z5LN2ixPcDAopT",temperature=0,model="gpt-4"),toolkit=toolkit,max_iterations=1000,verbose=True)
25
+ print(agent.run("what are the request method for openapi endpoint"))