getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-16 23:01:56 +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

2
packages/agent/src/agent-loop.ts
View file

@ -211,7 +211,7 @@ async function streamAssistantResponse(
const resolvedApiKey =
(config.getApiKey ? await config.getApiKey(config.model.provider) : undefined) || config.apiKey;
const response = streamFunction(config.model, llmContext, {
const response = await streamFunction(config.model, llmContext, {
...config,
apiKey: resolvedApiKey,
signal,

5
packages/agent/src/types.ts
View file

@ -11,7 +11,10 @@ import type {
} from "@mariozechner/pi-ai";
import type { Static, TSchema } from "@sinclair/typebox";
export type StreamFn = typeof streamSimple;
/** Stream function - can return sync or Promise for async config lookup */
export type StreamFn = (
...args: Parameters<typeof streamSimple>
) => ReturnType<typeof streamSimple> | Promise<ReturnType<typeof streamSimple>>;
/**
* Configuration for the agent loop.