Spaces:
Sleeping
Sleeping
Commit
·
d2d9578
1
Parent(s):
4f580f8
Update the ai agent to use deepseek model
Browse files- Dockerfile +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
|
99 |
-
ollama pull
|
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 |
-
|
48 |
-
self.BASE_URL = 'https://
|
49 |
self.headers = {
|
50 |
'Content-type': 'application/json'
|
51 |
}
|
52 |
-
self.user_id = '
|
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 |
|