fix: restore EMBED_URL pointing to ollama on 192.168.50.210:11434
This commit is contained in:
5
rag.py
5
rag.py
@@ -7,11 +7,12 @@ import httpx
|
||||
|
||||
from db import get_db, get_setting, list_skills_with_state, format_active_skills_prompt
|
||||
from memory import search_memories
|
||||
from config import LLAMA_SERVER_BASE, MAX_SKILL_PROMPT_CHARS
|
||||
from config import MAX_SKILL_PROMPT_CHARS
|
||||
|
||||
log = logging.getLogger("jarvischat")
|
||||
|
||||
QDRANT_URL = "http://192.168.50.108:6333"
|
||||
EMBED_URL = "http://192.168.50.210:11434"
|
||||
EMBED_MODEL = "mxbai-embed-large"
|
||||
RAG_COLLECTION = "jarvis_rag"
|
||||
RAG_SCORE_THRESHOLD = 0.25
|
||||
@@ -21,7 +22,7 @@ async def query_rag(query: str, limit: int = 3) -> list:
|
||||
try:
|
||||
async with httpx.AsyncClient() as client:
|
||||
embed_resp = await client.post(
|
||||
f"{LLAMA_SERVER_BASE}/api/embeddings",
|
||||
f"{EMBED_URL}/api/embeddings",
|
||||
json={"model": EMBED_MODEL, "prompt": query},
|
||||
timeout=10.0,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user