Skip to content

Commit 77156d0

Browse files
committed
Rebuild with deepseek install
1 parent 0215a54 commit 77156d0

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

shuffle-ai/1.0.0/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ RUN ls -alh /usr/bin
6565
RUN which ollama
6666

6767
#RUN /usr/local/bin/ollama pull llama3.2
68-
RUN ollama serve & sleep 2 && ollama pull llama3
68+
RUN ollama serve & sleep 2 && ollama pull nezahatkorkmaz/deepseek-v3
6969

7070
#RUN rm /usr/local/bin/ollama
7171
#RUN cd tesseract && ./autogen.sh && ./configure --build=x86_64-alpine-linux-musl --host=x86_64-alpine-linux-musl && make && make install && cd /tmp/src

shuffle-ai/1.0.0/api.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,13 @@ actions:
2121
example: ""
2222
schema:
2323
type: string
24+
- name: model
25+
description: "The model to run"
26+
required: false
27+
multiline: false
28+
example: "deepseek-v3"
29+
schema:
30+
type: string
2431

2532
- name: shuffle_cloud_inference
2633
description: Input ANY kind of data in the format you want, and the format you want it in. Default is a business-y email. Uses ShuffleGPT, which is based on OpenAI and our own model.

shuffle-ai/1.0.0/src/app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@ class Tools(AppBase):
3131
def __init__(self, redis, logger, console_logger=None):
3232
super().__init__(redis, logger, console_logger)
3333

34-
def run_llm(self, question, model="llama3.2"):
35-
models = []
34+
#def run_llm(self, question, model="llama3.2"):
35+
def run_llm(self, question, model="deepseek-v3"):
3636
response = ollama.chat(model=model, messages=[
3737
{
3838
"role": "user", "content": question,

0 commit comments

Comments
 (0)