summaryrefslogtreecommitdiff
path: root/rag/llm
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-06 01:40:19 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-06 01:40:19 +0200
commit7685677668e0e6987b582f8350c0573ed4a7abf6 (patch)
treedea8c845ec2d769b7f6d72cc59e3a79ea00a4904 /rag/llm
parent3a4efdbb50add7abd26a0f6e145dec1e0b897e77 (diff)
Format
Diffstat (limited to 'rag/llm')
-rw-r--r--rag/llm/encoder.py2
-rw-r--r--rag/llm/generator.py1
2 files changed, 1 insertions, 2 deletions
diff --git a/rag/llm/encoder.py b/rag/llm/encoder.py
index 94d5559..7c03dc5 100644
--- a/rag/llm/encoder.py
+++ b/rag/llm/encoder.py
@@ -1,5 +1,5 @@
import os
-from typing import List, Optional
+from typing import List
from uuid import uuid4
import ollama
diff --git a/rag/llm/generator.py b/rag/llm/generator.py
index 5f164e7..b0c6c40 100644
--- a/rag/llm/generator.py
+++ b/rag/llm/generator.py
@@ -31,5 +31,4 @@ class Generator:
def generate(self, prompt: Prompt, role: str) -> str:
log.debug("Generating answer...")
metaprompt = self.__metaprompt(role, prompt)
- print(f"metaprompt = \n{metaprompt}")
return ollama.generate(model=self.model, prompt=metaprompt)