diff options
Diffstat (limited to 'rag')
| -rw-r--r-- | rag/llm/encoder.py | 2 | ||||
| -rw-r--r-- | rag/llm/generator.py | 1 |
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) |