Merge pull request #39 from DefangLabs/jordan/devex2

A few more developer experience creature comforts

Author: jordanstephens

.devcontainer/Dockerfile

@@ -1 +1 @@
-FROM mcr.microsoft.com/devcontainers/python:3.12-bookworm
+FROM mcr.microsoft.com/devcontainers/python:3.9-bookworm

.devcontainer/devcontainer.json

@@ -6,5 +6,12 @@
   "features": {
     "ghcr.io/defanglabs/devcontainer-feature/defang-cli:1.0.4": {},
     "ghcr.io/devcontainers/features/docker-in-docker:2": {}
-  }
-}
+  },
+  "mounts": [
+    "source=/Users/jordan/.aws,target=/home/vscode/.aws,type=bind,consistency=cached",
+  ],
+  "forwardPorts": [
+    5050,
+    8080
+  ]
+}

.vscode/launch.json

@@ -0,0 +1,32 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python Debugger: Flask",
+            "type": "debugpy",
+            "request": "launch",
+            "module": "flask",
+            "cwd": "${workspaceFolder}/app",
+            "env": {
+                "FLASK_APP": "app.py",
+                "FLASK_DEBUG": "1",
+                "SECRET_KEY": "supersecret",
+                "SESSION_COOKIE_SECURE": "0",
+                "OPENAI_BASE_URL": "http://localhost:8080/api/v1",
+                "OPENAI_API_KEY": "secret"
+            },
+            "args": [
+                "run",
+                "--no-reload",
+                "--host=0.0.0.0",
+                "--port=5050"
+            ],
+            "justMyCode": false,
+            "jinja": true,
+            "autoStartBrowser": false
+        }
+    ]
+}

app/app.py

@@ -6,6 +6,7 @@
 import os
 import segment.analytics as analytics
 import uuid
+import sys
 
 analytics.write_key = os.getenv('SEGMENT_WRITE_KEY')
 
@@ -53,7 +54,7 @@ def generate():
                 yield token
                 full_response += token
         except Exception as e:
-            print(f"Error in /ask endpoint: {e}")
+            print(f"Error in /ask endpoint: {e}", file=sys.stderr)
             yield "Internal Server Error"
 
         if not full_response:

app/rag_system.py

@@ -1,6 +1,7 @@
 import openai
 import json
 import os
+import sys
 from datetime import date
 from sentence_transformers import SentenceTransformer
 import numpy as np
@@ -155,7 +156,7 @@ def answer_query_stream(self, query):
             self.conversation_history.append({"role": "assistant", "content": full_response})
 
         except Exception as e:
-            print(f"Error in answer_query_stream: {e}")
+            print(f"Error in answer_query_stream: {e}", file=sys.stderr)
             yield "An error occurred while generating the response."
 
     def clear_conversation_history(self):