Changed gemini system prompt integration

app.py

@@ -162,16 +162,21 @@ async def stream_response_gemini(query, history):
         f"CV: {full_cv_text}"
     )
 
-    # Build messages for Gemini
-    messages = [types.Content(role="system", parts=[types.Part(text=system_prompt)])]
+    # Build messages for Gemini (no system role - embed instructions in first user message)
+    messages = []
     
     # Add conversation history
     for msg in history:
         role = "user" if msg["role"] == "user" else "model"
-        messages.append(types.Content(role=role, parts=[types.Part(text=msg["content"])]))
+        messages.append(types.Content(role=role, parts=[types.Part.from_text(text=msg["content"])]))
     
-    # Add current query
-    messages.append(types.Content(role="user", parts=[types.Part(text=query)]))
+    # Add current query with system prompt embedded
+    if not history:  # If no history, include system prompt with the first message
+        combined_query = f"{system_prompt}\n\nUser question: {query}"
+    else:
+        combined_query = query
+    
+    messages.append(types.Content(role="user", parts=[types.Part.from_text(text=combined_query)]))
 
     try:
         response = gemini_client.models.generate_content_stream(
@@ -181,6 +186,7 @@ async def stream_response_gemini(query, history):
                 temperature=0.3,
                 top_p=0.7,
                 max_output_tokens=512,
+                response_mime_type="text/plain",
             )
         )
         

test_gemini_integration.py

@@ -30,6 +30,10 @@ class MockTypes:
     class Part:
         def __init__(self, text):
             self.text = text
+        
+        @classmethod
+        def from_text(cls, text):
+            return cls(text)
     
     class GenerateContentConfig:
         def __init__(self, temperature, top_p, max_output_tokens):
@@ -73,16 +77,21 @@ async def test_gemini_integration():
         f"CV: {full_cv_text}"
     )
 
-    # Build messages for Gemini
-    messages = [types.Content(role="system", parts=[types.Part(text=system_prompt)])]
+    # Build messages for Gemini (no system role - embed instructions in first user message)
+    messages = []
     
     # Add conversation history
     for msg in history:
         role = "user" if msg["role"] == "user" else "model"
-        messages.append(types.Content(role=role, parts=[types.Part(text=msg["content"])]))
+        messages.append(types.Content(role=role, parts=[types.Part.from_text(text=msg["content"])]))
     
-    # Add current query
-    messages.append(types.Content(role="user", parts=[types.Part(text=query)]))
+    # Add current query with system prompt embedded
+    if not history:  # If no history, include system prompt with the first message
+        combined_query = f"{system_prompt}\n\nUser question: {query}"
+    else:
+        combined_query = query
+    
+    messages.append(types.Content(role="user", parts=[types.Part.from_text(text=combined_query)]))
 
     print(f"System prompt length: {len(system_prompt)}")
     print(f"Number of messages: {len(messages)}")