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