abdibrahem commited on
Commit
d2d9578
·
1 Parent(s): 4f580f8

Update the ai agent to use deepseek model

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. main.py +4 -4
Dockerfile CHANGED
@@ -95,8 +95,8 @@ while ! curl -s http://localhost:11434 >/dev/null; do\n\
95
  sleep 2\n\
96
  done\n\
97
  \n\
98
- echo "Pulling Mistral model..."\n\
99
- ollama pull mistral\n\
100
  \n\
101
  echo "Starting FastAPI application..."\n\
102
  exec uvicorn main:app --host 0.0.0.0 --port 7860' > start.sh && \
 
95
  sleep 2\n\
96
  done\n\
97
  \n\
98
+ echo "Pulling Deepseek model..."\n\
99
+ ollama pull deepseek/deepseek-llm:7b\n\
100
  \n\
101
  echo "Starting FastAPI application..."\n\
102
  exec uvicorn main:app --host 0.0.0.0 --port 7860' > start.sh && \
main.py CHANGED
@@ -43,13 +43,13 @@ class AIAgent:
43
  def __init__(self):
44
  self.endpoints_documentation = endpoints_documentation
45
  self.ollama_base_url = "http://localhost:11434" # Default Ollama URL
46
- self.model_name = "mistral" # Using mistral model for better multilingual support
47
- # self.model_name = 'llama3'
48
- self.BASE_URL = 'https://8d30-105-196-69-205.ngrok-free.app'
49
  self.headers = {
50
  'Content-type': 'application/json'
51
  }
52
- self.user_id = '3b0b698d-ae49-4ba3-b83b-2b51fce7331d'
53
  self.max_retries = 3
54
  self.retry_delay = 2 # seconds
55
 
 
43
  def __init__(self):
44
  self.endpoints_documentation = endpoints_documentation
45
  self.ollama_base_url = "http://localhost:11434" # Default Ollama URL
46
+ # self.model_name = "mistral" # Using mistral model for better multilingual support
47
+ self.model_name = 'deepseek/deepseek-llm:7b'
48
+ self.BASE_URL = 'https://0a84-105-196-69-205.ngrok-free.app'
49
  self.headers = {
50
  'Content-type': 'application/json'
51
  }
52
+ self.user_id = '03564fa7-0827-43d5-ba1d-cff2177e2d09'
53
  self.max_retries = 3
54
  self.retry_delay = 2 # seconds
55