| import gradio as gr |
| from langchain import hub |
| from langchain_google_genai import ChatGoogleGenerativeAI |
| from langchain_core.messages import SystemMessage |
| from langchain.agents import create_react_agent, AgentExecutor |
| from langchain_community.agent_toolkits.load_tools import load_tools |
| from dotenv import find_dotenv, load_dotenv |
| import os |
|
|
| load_dotenv(find_dotenv()) |
| google_api_key = os.environ['GOOGLE_AI_API_KEY'] |
|
|
| prompt = hub.pull("hwchase17/react-chat") |
|
|
| template = """ |
| You are Codora, an AI assistant that helps the user with his/her code. |
| You accept user input and provide assistance in coding. DO NOT PROVIDE ANSWERS, ONLY GUIDE THEM TO THE SOLUTION. |
| Provide any necessary information to the user to help them understand the problem and solve it on their own. |
| Your answers should be crisp, clear and concise. |
| Use the provided tool to search Stack Overflow for relevant information. |
| Answer questions that are only related to coding/programming. |
| Don't respond to requests that are irrelevant to coding/programming. |
| Politely decline any irrelevant questions and remind them that you need a coding/programming request. |
| """ |
|
|
| messages = [SystemMessage(content=template)] |
|
|
| model = ChatGoogleGenerativeAI(model="gemini-1.5-pro", |
| google_api_key=google_api_key, |
| temperature=0.4) |
|
|
| tools = load_tools(["stackexchange"]) |
|
|
| agent = create_react_agent(llm=model, |
| prompt=prompt, |
| tools=tools) |
|
|
| agent_executor = AgentExecutor(agent=agent, |
| verbose=True, |
| tools=tools) |
|
|
| def respond(message, history): |
| messages.append(message) |
| response = agent_executor.invoke({"input": messages, "chat_history" : history}) |
| return response['output'] |
|
|
| demo = gr.ChatInterface( |
| respond, |
| examples=["How do I use async/await in a JavaScript function?", "How do I center a div in CSS?", "How do I create a virtual environment in Python?"] |
| ) |
|
|
| if __name__ == "__main__": |
| demo.launch() |
