Spaces:
Sleeping
Sleeping
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))
|