getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-16 01:03:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow async streamFn for dynamic proxy settings

Browse source 
- StreamFn type now allows returning Promise
- agent-loop awaits streamFn result
- createStreamFn takes getProxyUrl callback, reads settings on each call
This commit is contained in:
Mario Zechner 2025-12-28 11:39:46 +01:00
parent e0be2e650d
commit 92898f486b
4 changed files with 16 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

11
packages/web-ui/example/src/main.ts
View file

@ -161,10 +161,6 @@ const createAgent = async (initialState?: Partial<AgentState>) => {
agentUnsubscribe();
}
// Read proxy settings for streamFn
const proxyEnabled = await storage.settings.get<boolean>("proxy.enabled");
const proxyUrl = proxyEnabled ? (await storage.settings.get<string>("proxy.url")) || undefined : undefined;
agent = new Agent({
initialState: initialState || {
systemPrompt: `You are a helpful AI assistant with access to various tools.
@ -186,8 +182,11 @@ Feel free to use these tools when needed to provide accurate and helpful respons
const key = await storage.providerKeys.get(provider);
return key ?? undefined;
},
// Use streamFn with CORS proxy support
streamFn: createStreamFn(proxyUrl),
// Use streamFn with CORS proxy support (reads settings on each call)
streamFn: createStreamFn(async () => {
const enabled = await storage.settings.get<boolean>("proxy.enabled");
return enabled ? (await storage.settings.get<string>("proxy.url")) || undefined : undefined;
}),
});
agentUnsubscribe = agent.subscribe((event: any) => {