Spaces:
Running
Running
Update aiEngine.js
Browse files- aiEngine.js +7 -3
aiEngine.js
CHANGED
|
@@ -29,7 +29,7 @@ export const AIEngine = {
|
|
| 29 |
};
|
| 30 |
|
| 31 |
const contents = [
|
| 32 |
-
|
| 33 |
...history,
|
| 34 |
{ role: 'user', parts: [{ text: input }] }
|
| 35 |
];
|
|
@@ -51,9 +51,13 @@ export const AIEngine = {
|
|
| 51 |
* WORKER MODEL (Gemini 2.5 Flash - Fast execution)
|
| 52 |
*/
|
| 53 |
callWorker: async (history, input, imagePart = null) => {
|
| 54 |
-
|
| 55 |
const config = {
|
| 56 |
thinkingConfig: { thinkingBudget: -1 }, // Standard generation
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
tools: [{
|
| 58 |
systemInstruction: [{
|
| 59 |
text: prompts.worker_system_prompt
|
|
@@ -67,7 +71,7 @@ export const AIEngine = {
|
|
| 67 |
}
|
| 68 |
|
| 69 |
const contents = [
|
| 70 |
-
|
| 71 |
...history,
|
| 72 |
{ role: 'user', parts: currentParts }
|
| 73 |
];
|
|
|
|
| 29 |
};
|
| 30 |
|
| 31 |
const contents = [
|
| 32 |
+
{ role: 'user', parts: [{ text: prompts.pm_system_prompt }] }, // System instruction injection
|
| 33 |
...history,
|
| 34 |
{ role: 'user', parts: [{ text: input }] }
|
| 35 |
];
|
|
|
|
| 51 |
* WORKER MODEL (Gemini 2.5 Flash - Fast execution)
|
| 52 |
*/
|
| 53 |
callWorker: async (history, input, imagePart = null) => {
|
| 54 |
+
/* const modelId = 'gemini-flash-latest'; // Per prompt requirements
|
| 55 |
const config = {
|
| 56 |
thinkingConfig: { thinkingBudget: -1 }, // Standard generation
|
| 57 |
+
*/
|
| 58 |
+
const modelId = 'gemini-3-pro-preview'; // Per prompt requirements
|
| 59 |
+
const config = {
|
| 60 |
+
thinkingConfig: { thinkingLevel: 'HIGH' },
|
| 61 |
tools: [{
|
| 62 |
systemInstruction: [{
|
| 63 |
text: prompts.worker_system_prompt
|
|
|
|
| 71 |
}
|
| 72 |
|
| 73 |
const contents = [
|
| 74 |
+
{ role: 'user', parts: [{ text: prompts.worker_system_prompt }] },
|
| 75 |
...history,
|
| 76 |
{ role: 'user', parts: currentParts }
|
| 77 |
];
|