Spaces:
Runtime error
Runtime error
| import os | |
| from langchain.agents import Tool | |
| from langchain.memory import ConversationBufferMemory | |
| from langchain.chat_models import ChatOpenAI | |
| from langchain.utilities import GoogleSearchAPIWrapper | |
| from langchain.agents import initialize_agent | |
| import gradio as gr | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| #OPENAI_API_KEY =os.getenv("OPENAI_API_KEY") | |
| GOOGLE_API_KEY =os.getenv("GOOGLE_API_KEY") | |
| GOOGLE_CSE_ID =os.getenv("GOOGLE_CSE_ID") | |
| repo_id=os.getenv("repo_id") | |
| search = GoogleSearchAPIWrapper() | |
| tools = [ | |
| Tool( | |
| name ="Search" , | |
| func=search.run, | |
| description="To be used when you need to answer questions about current events or you are not sure of an answer to user question." | |
| ), | |
| ] | |
| memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True) | |
| llm=ChatOpenAI(temperature=0) | |
| #llm = OpenAI() | |
| #llm = OpenAI() | |
| agent_chain = initialize_agent(tools, llm, agent="chat-conversational-react-description", | |
| verbose=True, memory=memory) | |
| def chat_response(input_text): | |
| response = agent_chain.run(input=input_text) | |
| return response | |
| interface = gr.Interface(fn=chat_response, inputs="text", outputs="text", description="Chat with a conversational agent") | |
| interface.launch() |