Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -197,3 +197,11 @@ report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json

# QEMU
*.qcow2

docker/start-proxy.bat
docker/start.bat
docker/stop-proxy.bat
docker/stop.bat
packages/bytebot-agent-build.bat
packages/Dockerfile
.prettierrc
2 changes: 2 additions & 0 deletions docker/docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@ services:
- ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
- OPENAI_API_KEY=${OPENAI_API_KEY}
- GEMINI_API_KEY=${GEMINI_API_KEY}
- DOUBAO_API_KEY=${DOUBAO_API_KEY}
- DOUBAO_BASE_URL=${DOUBAO_BASE_URL}
depends_on:
- postgres
networks:
Expand Down
5 changes: 3 additions & 2 deletions packages/bytebot-agent/.prettierrc
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
{
"singleQuote": true,
"trailingComma": "all"
}
"trailingComma": "all",
"endOfLine": "auto"
}
2 changes: 2 additions & 0 deletions packages/bytebot-agent/src/agent/agent.module.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import { GoogleModule } from '../google/google.module';
import { SummariesModule } from 'src/summaries/summaries.modue';
import { AgentAnalyticsService } from './agent.analytics';
import { ProxyModule } from 'src/proxy/proxy.module';
import { DoubaoModule } from '../doubao/doubao.module';

@Module({
imports: [
Expand All @@ -22,6 +23,7 @@ import { ProxyModule } from 'src/proxy/proxy.module';
OpenAIModule,
GoogleModule,
ProxyModule,
DoubaoModule,
],
providers: [
AgentProcessor,
Expand Down
3 changes: 3 additions & 0 deletions packages/bytebot-agent/src/agent/agent.processor.ts
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ import {
import { SummariesService } from '../summaries/summaries.service';
import { handleComputerToolUse } from './agent.computer-use';
import { ProxyService } from '../proxy/proxy.service';
import { DoubaoService } from '../doubao/doubao.service';

@Injectable()
export class AgentProcessor {
Expand All @@ -57,12 +58,14 @@ export class AgentProcessor {
private readonly googleService: GoogleService,
private readonly proxyService: ProxyService,
private readonly inputCaptureService: InputCaptureService,
private readonly doubaoService: DoubaoService,
) {
this.services = {
anthropic: this.anthropicService,
openai: this.openaiService,
google: this.googleService,
proxy: this.proxyService,
doubao: this.doubaoService,
};
this.logger.log('AgentProcessor initialized');
}
Expand Down
2 changes: 1 addition & 1 deletion packages/bytebot-agent/src/agent/agent.types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ export interface BytebotAgentService {
}

export interface BytebotAgentModel {
provider: 'anthropic' | 'openai' | 'google' | 'proxy';
provider: 'anthropic' | 'openai' | 'google' | 'proxy' | 'doubao';
name: string;
title: string;
contextWindow?: number;
Expand Down
12 changes: 12 additions & 0 deletions packages/bytebot-agent/src/doubao/doubao.constants.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
import { BytebotAgentModel } from 'src/agent/agent.types';

export const DOUBAO_MODELS: BytebotAgentModel[] = [
{
provider: 'doubao',
name: 'doubao-seed-1-6-vision-250815',
title: 'doubao-seed-1-6-vision',
contextWindow: 1047576,
},
];

export const DEFAULT_MODEL = DOUBAO_MODELS[0];
10 changes: 10 additions & 0 deletions packages/bytebot-agent/src/doubao/doubao.module.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
import { Module } from '@nestjs/common';
import { ConfigModule } from '@nestjs/config';
import { DoubaoService } from './doubao.service';

@Module({
imports: [ConfigModule],
providers: [DoubaoService],
exports: [DoubaoService],
})
export class DoubaoModule {}
Loading