feat: migrate to Letta TS SDK v1 (alpha) (#11)
This commit is contained in:
@@ -2,22 +2,27 @@
|
||||
* Utilities for sending messages to an agent
|
||||
**/
|
||||
|
||||
import type { Letta } from "@letta-ai/letta-client";
|
||||
import type { Stream } from "@letta-ai/letta-client/core/streaming";
|
||||
import type { MessageCreate } from "@letta-ai/letta-client/resources/agents/agents";
|
||||
import type {
|
||||
ApprovalCreate,
|
||||
LettaStreamingResponse,
|
||||
} from "@letta-ai/letta-client/resources/agents/messages";
|
||||
import { getClient } from "./client";
|
||||
|
||||
export async function sendMessageStream(
|
||||
agentId: string,
|
||||
messages: Array<Letta.MessageCreate | Letta.ApprovalCreate>,
|
||||
messages: Array<MessageCreate | ApprovalCreate>,
|
||||
opts: {
|
||||
streamTokens?: boolean;
|
||||
background?: boolean;
|
||||
// add more later: includePings, request timeouts, etc.
|
||||
} = { streamTokens: true, background: true },
|
||||
): Promise<AsyncIterable<Letta.LettaStreamingResponse>> {
|
||||
): Promise<Stream<LettaStreamingResponse>> {
|
||||
const client = await getClient();
|
||||
return client.agents.messages.createStream(agentId, {
|
||||
return client.agents.messages.stream(agentId, {
|
||||
messages: messages,
|
||||
streamTokens: opts.streamTokens ?? true,
|
||||
stream_tokens: opts.streamTokens ?? true,
|
||||
background: opts.background ?? true,
|
||||
});
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user