Spaces:
Running
Running
Update app/rag.py
Browse files- app/rag.py +7 -1
app/rag.py
CHANGED
@@ -6,6 +6,12 @@ import time
|
|
6 |
|
7 |
load_dotenv()
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
logger = logging.getLogger(__name__)
|
10 |
|
11 |
# Define the RAG prompt template
|
@@ -30,7 +36,7 @@ Guidelines:
|
|
30 |
class RAGSystem:
|
31 |
def __init__(self, data_processor):
|
32 |
self.data_processor = data_processor
|
33 |
-
self.model = os.getenv('OLLAMA_MODEL', 'phi3')
|
34 |
self.ollama_host = os.getenv('OLLAMA_HOST', 'http://ollama:11434')
|
35 |
self.timeout = int(os.getenv('OLLAMA_TIMEOUT', 240))
|
36 |
self.max_retries = int(os.getenv('OLLAMA_MAX_RETRIES', 3))
|
|
|
6 |
|
7 |
load_dotenv()
|
8 |
|
9 |
+
# Configure logging for stdout only
|
10 |
+
logging.basicConfig(
|
11 |
+
level=logging.INFO,
|
12 |
+
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
13 |
+
stream=sys.stdout
|
14 |
+
)
|
15 |
logger = logging.getLogger(__name__)
|
16 |
|
17 |
# Define the RAG prompt template
|
|
|
36 |
class RAGSystem:
|
37 |
def __init__(self, data_processor):
|
38 |
self.data_processor = data_processor
|
39 |
+
self.model = os.getenv('OLLAMA_MODEL', 'phi3.5')
|
40 |
self.ollama_host = os.getenv('OLLAMA_HOST', 'http://ollama:11434')
|
41 |
self.timeout = int(os.getenv('OLLAMA_TIMEOUT', 240))
|
42 |
self.max_retries = int(os.getenv('OLLAMA_MAX_RETRIES', 3))
|