danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use /no_think for Ollama

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-08-15 23:13:27 +02:00
parent 79841f32ca
commit 60543caa36
No known key found for this signature in database
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/haiku/rag/qa/prompts.py
View file

@ -18,6 +18,7 @@ Guidelines:
- Stick to the answer, do not ellaborate or provide context unless explicitly asked for it.
Be concise, and always maintain accuracy over completeness. Prefer short, direct answers that are well-supported by the documents.
/no_think
"""
SYSTEM_PROMPT_WITH_CITATIONS = """
@ -55,4 +56,5 @@ Citations:
- /path/to/document2.pdf: "The manual provides guidance on military procedures and..."
Be concise, and always maintain accuracy over completeness. Prefer short, direct answers that are well-supported by the documents.
/no_think
"""

4
src/haiku/rag/reranking/ollama.py
View file

@ -32,7 +32,9 @@ Return your response as a JSON object with a "results" array. Each result should
- "index": the original index of the document (integer)
- "relevance_score": a score between 0.0 and 1.0 indicating relevance (float, where 1.0 is most relevant)
Only return the top documents up to the requested limit, ordered by decreasing relevance score."""
Only return the top documents up to the requested limit, ordered by decreasing relevance score.
/no_think
"""
model_obj = OpenAIModel(
model_name=model,