gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Increase the machine size

Browse source
This commit is contained in:
Pavel Zorin 2025-12-01 15:21:58 +01:00
parent 7c9a78abea
commit ba9ca46574
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
.github/workflows/test_ollama.yml vendored
View file

@ -7,8 +7,8 @@ jobs:
run_ollama_test: run_ollama_test:
# needs 32 Gb RAM for phi4 in a container
runs-on: buildjet-4vcpu-ubuntu-2204 runs-on: buildjet-8vcpu-ubuntu-2204
steps: steps:
- name: Checkout repository - name: Checkout repository
@ -47,15 +47,15 @@ jobs:
- name: Pull required Ollama models - name: Pull required Ollama models
run: | run: |
curl -X POST http://localhost:11434/api/pull -d '{"name": "phi3:mini"}' curl -X POST http://localhost:11434/api/pull -d '{"name": "phi4"}'
curl -X POST http://localhost:11434/api/pull -d '{"name": "nomic-embed-text"}' curl -X POST http://localhost:11434/api/pull -d '{"name": "avr/sfr-embedding-mistral:latest"}'
- name: Call ollama API - name: Call ollama API
run: | run: |
curl -X POST http://localhost:11434/v1/chat/completions \ curl -X POST http://localhost:11434/v1/chat/completions \
-H "Content-Type: application/json" \ -H "Content-Type: application/json" \
-d '{ -d '{
"model": "phi3:mini", "model": "phi4",
"stream": false, "stream": false,
"messages": [ "messages": [
{ "role": "system", "content": "You are a helpful assistant." }, { "role": "system", "content": "You are a helpful assistant." },
@ -65,7 +65,7 @@ jobs:
curl -X POST http://127.0.0.1:11434/api/embed \ curl -X POST http://127.0.0.1:11434/api/embed \
-H "Content-Type: application/json" \ -H "Content-Type: application/json" \
-d '{ -d '{
"model": "nomic-embed-text", "model": "avr/sfr-embedding-mistral:latest",
"input": "This is a test sentence to generate an embedding." "input": "This is a test sentence to generate an embedding."
}' }'
@ -82,10 +82,10 @@ jobs:
LLM_PROVIDER: "ollama" LLM_PROVIDER: "ollama"
LLM_API_KEY: "ollama" LLM_API_KEY: "ollama"
LLM_ENDPOINT: "http://localhost:11434/v1/" LLM_ENDPOINT: "http://localhost:11434/v1/"
LLM_MODEL: "phi3:mini" LLM_MODEL: "phi4"
EMBEDDING_PROVIDER: "ollama" EMBEDDING_PROVIDER: "ollama"
EMBEDDING_MODEL: "nomic-embed-text" EMBEDDING_MODEL: "avr/sfr-embedding-mistral:latest"
EMBEDDING_ENDPOINT: "http://localhost:11434/api/embed" EMBEDDING_ENDPOINT: "http://localhost:11434/api/embed"
EMBEDDING_DIMENSIONS: "768" EMBEDDING_DIMENSIONS: "4096"
HUGGINGFACE_TOKENIZER: "nomic-ai/nomic-embed-text-v1" HUGGINGFACE_TOKENIZER: "Salesforce/SFR-Embedding-Mistral"
run: uv run python ./examples/python/simple_example.py run: uv run python ./examples/python/simple_example.py