Skip to content

Commit 6256998

Browse files
committed
patch gemini, webgenui, generic aibitat providers + disable providers unable to test
1 parent 2d0768b commit 6256998

File tree

9 files changed

+10
-10
lines changed

9 files changed

+10
-10
lines changed

server/utils/agents/aibitat/providers/azure.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class AzureOpenAiProvider extends Provider {
2020
}
2121

2222
get supportsAgentStreaming() {
23-
return true;
23+
return false;
2424
}
2525

2626
/**

server/utils/agents/aibitat/providers/dellProAiStudio.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ class DellProAiStudioProvider extends InheritMultiple([Provider, UnTooled]) {
3434
}
3535

3636
get supportsAgentStreaming() {
37-
return true;
37+
return false;
3838
}
3939

4040
async #handleFunctionCallChat({ messages = [] }) {

server/utils/agents/aibitat/providers/gemini.js

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ class GeminiProvider extends InheritMultiple([Provider, UnTooled]) {
3535
}
3636

3737
get supportsAgentStreaming() {
38-
return true;
38+
return false;
3939
}
4040

4141
/**
@@ -206,7 +206,7 @@ class GeminiProvider extends InheritMultiple([Provider, UnTooled]) {
206206
// _but_ we should enable it to call previously used tools in a new chat interaction.
207207
this.deduplicator.reset("runs");
208208
return {
209-
result: completion.content,
209+
textResponse: completion.content,
210210
cost: 0,
211211
};
212212
} catch (error) {

server/utils/agents/aibitat/providers/genericOpenAi.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ class GenericOpenAiProvider extends InheritMultiple([Provider, UnTooled]) {
102102
// _but_ we should enable it to call previously used tools in a new chat interaction.
103103
this.deduplicator.reset("runs");
104104
return {
105-
result: completion.content,
105+
textResponse: completion.content,
106106
cost: 0,
107107
};
108108
} catch (error) {

server/utils/agents/aibitat/providers/localai.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ class LocalAiProvider extends InheritMultiple([Provider, UnTooled]) {
2828
}
2929

3030
get supportsAgentStreaming() {
31-
return true;
31+
return false;
3232
}
3333

3434
async #handleFunctionCallChat({ messages = [] }) {

server/utils/agents/aibitat/providers/nvidiaNim.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class NvidiaNimProvider extends InheritMultiple([Provider, UnTooled]) {
2929
}
3030

3131
get supportsAgentStreaming() {
32-
return true;
32+
return false;
3333
}
3434

3535
async #handleFunctionCallChat({ messages = [] }) {

server/utils/agents/aibitat/providers/ppio.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ class PPIOProvider extends InheritMultiple([Provider, UnTooled]) {
3232
}
3333

3434
get supportsAgentStreaming() {
35-
return true;
35+
return false;
3636
}
3737

3838
async #handleFunctionCallChat({ messages = [] }) {

server/utils/agents/aibitat/providers/textgenwebui.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ class TextWebGenUiProvider extends InheritMultiple([Provider, UnTooled]) {
1818
});
1919

2020
this._client = client;
21-
this.model = null; // text-web-gen-ui does not have a model pref.
21+
this.model = "text-generation-webui"; // text-web-gen-ui does not have a model pref, but we need a placeholder
2222
this.verbose = true;
2323
}
2424

server/utils/agents/index.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ class AgentHandler {
249249
case "perplexity":
250250
return process.env.PERPLEXITY_MODEL_PREF ?? "sonar-small-online";
251251
case "textgenwebui":
252-
return null;
252+
return "text-generation-webui";
253253
case "bedrock":
254254
return process.env.AWS_BEDROCK_LLM_MODEL_PREFERENCE ?? null;
255255
case "fireworksai":

0 commit comments

Comments
 (0)