test / app.py
uyen13's picture
Update app.py
0abbf38 verified
from langchain.chat_models import ChatOpenAI
from langchain.agents import create_json_agent
from langchain.agents.agent_toolkits import JsonToolkit
from langchain.tools.json.tool import JsonSpec
import json
# file="output.json"
# with open(file,"r") as f1:
# data=json.load(f1)
# f1.close()
data ={
"openapi": "3.0.0",
"info": {
"title": "OpenAI API",
"description": "The OpenAI REST API. Please see https://platform.openai.com/docs/api-reference for more details.",
"version": "2.0.0",
"termsOfService": "https://openai.com/policies/terms-of-use",
"contact": {
"name": "OpenAI Support",
"url": "https://help.openai.com/"
}}}
spec=JsonSpec(dict_=data,max_value_length=4000)
toolkit=JsonToolkit(spec=spec)
agent=create_json_agent(llm=ChatOpenAI(openai_api_key="sk-nAqoChT9cmkAxALwMLdWT3BIbkFJcNHsH5Z5LN2ixPcDAopT",temperature=0,model="gpt-4"),toolkit=toolkit,max_iterations=1000,verbose=True)
print(agent.run("what are the request method for openapi endpoint"))