From 73a4adb6d4161ea222ed1ba56413571892da73d6 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 9 Apr 2024 01:04:50 +0200
Subject: Fix imports

---
 rag/ui.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'rag/ui.py')

diff --git a/rag/ui.py b/rag/ui.py
index 37ab664..18e6bd1 100644
--- a/rag/ui.py
+++ b/rag/ui.py
@@ -4,8 +4,8 @@ from langchain_community.document_loaders.blob_loaders import Blob
 
 from dotenv import load_dotenv
 from rag.generator import get_generator, MODELS
-from generator.prompt import Prompt
-from retriever.retriever import Retriever
+from rag.generator.prompt import Prompt
+from rag.retriever.retriever import Retriever
 
 
 if __name__ == "__main__":
-- 
cgit v1.2.3-70-g09d2