Merge pull request #67 from DefangLabs/linda-gcp-gateway-edits

gcp gateway edits

Author: commit111

.gitignore

@@ -3,3 +3,4 @@ __pycache__
 sentence-transformers
 .tmp/
 node_modules
+venv

app/rag_system.py

@@ -148,12 +148,12 @@ def answer_query_stream(self, query):
             )
 
             collected_messages = []
-            for chunk in stream:
-                if chunk['choices'][0]['finish_reason'] is not None:
-                    break
+            for chunk in stream:  
                 content = chunk['choices'][0]['delta'].get('content', '')
                 collected_messages.append(content)
                 yield content
+                if chunk['choices'][0].get('finish_reason') is not None:
+                    break
 
             if len(citations) > 0:
                 yield "\n\nReferences:\n" + "\n".join(citations)