Update app.py

app.py

@@ -163,11 +163,81 @@ async def get_proxy():
              async callAgent(model, messages, temperature = 0.7) { await this.rateLimitWait(); const startTime = Date.now(); console.log("Calling Agent:", model); try { const response = await fetch(this.apiUrl + 'chat/completions', { method: 'POST', headers: { 'Content-Type': 'application/json', 'Authorization': `Bearer ${this.apiKey}` }, body: JSON.stringify({ model: model, messages: messages, temperature: temperature }) }); const endTime = Date.now(); this.callTimestamps.push(endTime); console.log(`API call took ${endTime - startTime} ms`); if (!response.ok) { const errorData = await response.json().catch(() => ({ error: { message: response.statusText } })); console.error("API Error:", errorData); throw new Error(errorData.error?.message || `API failed: ${response.status}`); } const data = await response.json(); if (!data.choices || !data.choices[0]?.message) throw new Error("Invalid API response structure"); console.log("API Response received."); return data.choices[0].message.content; } catch (error) { this.callTimestamps.push(Date.now()); console.error('Error calling agent:', error); throw error; } }
              setMaxCallsPerMinute(value) { const parsedValue = parseInt(value, 10); if (!isNaN(parsedValue) && parsedValue > 0) { console.log(`Max calls/min set to: ${parsedValue}`); this.maxCallsPerMinute = parsedValue; return true; } console.warn(`Invalid max calls/min: ${value}`); return false; }
          }
-         class ConversationalAgentClient extends BaseAgentClient {
-             constructor(apiKey, apiUrl = 'https://llm.synapse.thalescloud.io/v1/') { super(apiKey, apiUrl); }
-             async call(model, userPrompt, systemPrompt, conversationHistory = [], temperature = 0.7) { const messages = [{ role: 'system', content: systemPrompt }, ...conversationHistory, { role: 'user', content: userPrompt }]; const assistantResponse = await super.callAgent(model, messages, temperature); const updatedHistory = [...conversationHistory, { role: 'user', content: userPrompt }, { role: 'assistant', content: assistantResponse }]; return { response: assistantResponse, history: updatedHistory }; }
-             async callWithCodeContext(model, userPrompt, systemPrompt, selectedCodeVersionsData = [], conversationHistory = [], temperature = 0.7) { let codeContext = ""; let fullSystemPrompt = systemPrompt || ""; if (selectedCodeVersionsData && selectedCodeVersionsData.length > 0) { codeContext = "Code context (chronological):\n\n"; selectedCodeVersionsData.forEach((versionData, index) => { if (versionData && typeof versionData.code === 'string') codeContext += `--- Part ${index + 1} (${versionData.version || '?'}) ---\n${versionData.code}\n\n`; else console.warn(`Invalid context version data at index ${index}`); }); codeContext += "-------- end context ---\n\nUser request based on context:\n\n"; } const fullPrompt = codeContext + userPrompt; const messages = [{ role: 'system', content: fullSystemPrompt }, ...conversationHistory, { role: 'user', content: fullPrompt }]; const assistantResponse = await super.callAgent(model, messages, temperature); const updatedHistory = [...conversationHistory, { role: 'user', content: fullPrompt }, { role: 'assistant', content: assistantResponse }]; return { response: assistantResponse, history: updatedHistory }; }
-         }
+        class ConversationalAgentClient extends BaseAgentClient {
+            constructor(apiKey, apiUrl = 'https://llm.synapse.thalescloud.io/v1/') {
+                super(apiKey, apiUrl);
+            }
+        
+            async call(model, userPrompt, systemPrompt, conversationHistory = [], temperature = 0.7) {
+                const messages = [
+                    { role: 'system', content: systemPrompt },
+                    ...conversationHistory,
+                    { role: 'user', content: userPrompt }
+                ];
+                
+                const assistantResponse = await super.callAgent(model, messages, temperature);
+                
+                const updatedHistory = [
+                    ...conversationHistory,
+                    { role: 'user', content: userPrompt },
+                    { role: 'assistant', content: assistantResponse }
+                ];
+                
+                return { 
+                    response: assistantResponse, 
+                    history: updatedHistory 
+                };
+            }
+        
+            async callWithCodeContext(
+                model,
+                userPrompt,
+                systemPrompt,
+                selectedCodeVersionsData = [],
+                conversationHistory = [],
+                temperature = 0.7
+            ) {
+                let codeContext = "";
+                let fullSystemPrompt = systemPrompt || "";
+                
+                if (selectedCodeVersionsData && selectedCodeVersionsData.length > 0) {
+                    codeContext = "Code context (chronological):\n\n";
+                    
+                    selectedCodeVersionsData.forEach((versionData, index) => {
+                        if (versionData && typeof versionData.code === 'string') {
+                            codeContext += `--- Part ${index + 1} (${versionData.version || '?'}) ---\n`;
+                            codeContext += `${versionData.code}\n\n`;
+                        } else {
+                            console.warn(`Invalid context version data at index ${index}`);
+                        }
+                    });
+                    
+                    codeContext += "-------- end context ---\n\nUser request based on context:\n\n";
+                }
+                
+                const fullPrompt = codeContext + userPrompt;
+                
+                const messages = [
+                    { role: 'system', content: fullSystemPrompt },
+                    ...conversationHistory,
+                    { role: 'user', content: fullPrompt }
+                ];
+                
+                const assistantResponse = await super.callAgent(model, messages, temperature);
+                
+                const updatedHistory = [
+                    ...conversationHistory,
+                    { role: 'user', content: fullPrompt },
+                    { role: 'assistant', content: assistantResponse }
+                ];
+                
+                return { 
+                    response: assistantResponse, 
+                    history: updatedHistory 
+                };
+            }
+        }
+
         </script>
     </body>
     </html>