Nikolay Angelov commited on
Commit
4fc4736
·
1 Parent(s): 06972df

add docker compose

Browse files
Files changed (3) hide show
  1. .gitignore +2 -1
  2. app.py +3 -4
  3. docker-compose.yml +10 -0
.gitignore CHANGED
@@ -1,2 +1,3 @@
1
  .venv/
2
- __pycache__/
 
 
1
  .venv/
2
+ __pycache__/
3
+ .env
app.py CHANGED
@@ -3,7 +3,6 @@ from langchain_huggingface import HuggingFaceEndpoint
3
  from langchain.agents import AgentExecutor, create_react_agent
4
  from langchain_core.prompts import PromptTemplate
5
 
6
- from tools.final_answer import final_answer
7
  from tools.visit_webpage import visit_webpage
8
 
9
  import gradio as gr
@@ -16,9 +15,9 @@ import uuid
16
  from typing import Optional, Type, Dict, Any, List
17
  from pydantic import BaseModel, Field
18
 
19
- from fastapi import FastAPI, BackgroundTasks, HTTPException, Request
20
  from fastapi.middleware.cors import CORSMiddleware
21
- from fastapi.responses import JSONResponse, RedirectResponse
22
  import requests
23
 
24
  # Initialize FastAPI app
@@ -81,7 +80,7 @@ with open("prompts.yaml", 'r') as stream:
81
  prompt = PromptTemplate.from_template(prompt_templates["template"])
82
 
83
  # Create the agent
84
- tools = [get_current_time, visit_webpage] # Removed final_answer as it's handled by ReAct format
85
  agent = create_react_agent(
86
  llm=llm,
87
  tools=tools,
 
3
  from langchain.agents import AgentExecutor, create_react_agent
4
  from langchain_core.prompts import PromptTemplate
5
 
 
6
  from tools.visit_webpage import visit_webpage
7
 
8
  import gradio as gr
 
15
  from typing import Optional, Type, Dict, Any, List
16
  from pydantic import BaseModel, Field
17
 
18
+ from fastapi import FastAPI, HTTPException
19
  from fastapi.middleware.cors import CORSMiddleware
20
+ from fastapi.responses import RedirectResponse
21
  import requests
22
 
23
  # Initialize FastAPI app
 
80
  prompt = PromptTemplate.from_template(prompt_templates["template"])
81
 
82
  # Create the agent
83
+ tools = [get_current_time, visit_webpage]
84
  agent = create_react_agent(
85
  llm=llm,
86
  tools=tools,
docker-compose.yml ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ # docker-compose.yml
2
+ version: '3'
3
+ services:
4
+ app:
5
+ build: .
6
+ ports:
7
+ - "8000:8000"
8
+ - "7860:7860"
9
+ env_file:
10
+ - .env