add comment

DigitalProductInnovationAndDevelopment · Aug 7, 2024 · bd78ef8 · bd78ef8
1 parent 8ff85c7
commit bd78ef8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/.env.docker.example b/.env.docker.example
@@ -13,7 +13,7 @@ POSTGRES_PASSWORD=postgres
 
 REDIS_ENDPOINT=redis://redis:6379/0
 
-ENVIRONMENT=development
+ENVIRONMENT=production
 
 AI_STRATEGY=OLLAMA   #OLLAMA,OPENAI,ANTHROPIC
 

diff --git a/src/app.py b/src/app.py
@@ -48,8 +48,11 @@ def health():
         "uptime": round(time.time() - start_time, 2),
         "external_modules": {"ollama": ollama_health},
         "urls": {
-            # "llm": llm_service.get_url(),
-            "redis": config.redis_endpoint,
+            "redis": (
+                config.redis_endpoint
+                if config.environment == "development"
+                else "retracted"
+            ),
             # this leaks the db user and password in dev mode
             "postgres": (
                 config.get_db_url()

diff --git a/src/test/mock_llm.py b/src/test/mock_llm.py
@@ -58,7 +58,7 @@ def get_url(self) -> str:
         return self.generate_url
 
     def _generate(self, prompt: str, json=True) -> Dict[str, str]:
-        print(prompt)
+        # TODO: Mock Aggregated Solution Responses and Subdivision Responses
         print(f"{answer_in_json_prompt('recommendation')}".format())
         if f"{answer_in_json_prompt('combined_description').format()}" in prompt:
             return {"combined_description": "combined_description"}