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"))