diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-06 01:40:19 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-06 01:40:19 +0200 |
commit | 7685677668e0e6987b582f8350c0573ed4a7abf6 (patch) | |
tree | dea8c845ec2d769b7f6d72cc59e3a79ea00a4904 /rag/llm | |
parent | 3a4efdbb50add7abd26a0f6e145dec1e0b897e77 (diff) |
Format
Diffstat (limited to 'rag/llm')
-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) |