ganesh3 commited on
Commit
61c90c9
·
verified ·
1 Parent(s): 5de591d

Update app/rag.py

Browse files
Files changed (1) hide show
  1. 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))