From 7685677668e0e6987b582f8350c0573ed4a7abf6 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 6 Apr 2024 01:40:19 +0200
Subject: Format

---
 rag/llm/encoder.py   | 2 +-
 rag/llm/generator.py | 1 -
 2 files changed, 1 insertion(+), 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)
-- 
cgit v1.2.3-70-g09d2