thoristhor's picture
Update app.py
b24ce13
raw
history blame
1.28 kB
import os
import gradio as gr
import pinecone
from gpt_index import GPTIndexMemory, GPTPineconeIndex
from langchain.agents import Tool
from langchain.chains.conversation.memory import ConversationBufferMemory
from langchain import OpenAI
from langchain.agents import initialize_agent
OPENAI_API_KEY=os.environ.get("OPENAI_API_KEY")
PINECONE_API_KEY=os.environ.get("PINECONE_API_KEY")
PINECONE_ENV=os.environ.get("PINECONE_ENV")
pindex=pinecone.Index("sethgodin")
pinedex=GPTPineconeIndex([], pinecone_index=pindex)
tools = [
Tool(
name = "GPT Index",
func=lambda q: str(pinedex.query(q)),
description="useful for when you want to answer questions about the author. The input to this tool should be a complete english sentence.",
return_direct=True
),
]
memory = ConversationBufferMemory(memory_key="chat_history")
llm=OpenAI(temperature=0)
agent_chain = initialize_agent(tools, llm, agent="conversational-react-description", memory=memory)
def run_something(isfa: str):
return agent_chain.run(input=isfa)
with gr.Blocks() as demo:
chatbot = gr.Chatbot()
with gr.Row():
txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(container=False)
txt.submit(run_something)
demo.launch()