UNPKG

genkitx-patientseek

Version:

A community plugin for Firebase Genkit to integrate DeepSeek medical models.

14 lines (13 loc) 1.51 kB
import type { CandidateData } from "genkit/model"; import type { StreamingCallback, GenerateRequest, GenerateResponseData, MessageData, Role, ToolRequestPart } from "genkit"; import type { GenerateResponseChunkData } from "genkit/model"; import type OpenAI from "openai"; import { type ChatCompletion, type ChatCompletionChunk, type ChatCompletionMessageParam, type ChatCompletionMessageToolCall, type ChatCompletionRole } from "openai/resources/index.mjs"; import { DeepSeekConfigSchema } from "./models.js"; export declare function toOpenAIRole(role: Role): ChatCompletionRole; export declare function toOpenAiMessages(messages: MessageData[]): ChatCompletionMessageParam[]; export declare function fromOpenAiToolCall(toolCall: ChatCompletionMessageToolCall | ChatCompletionChunk.Choice.Delta.ToolCall, choice: ChatCompletion.Choice | ChatCompletionChunk.Choice): ToolRequestPart; export declare function fromOpenAiChoice(choice: ChatCompletion.Choice, jsonMode?: boolean): CandidateData; export declare function fromOpenAiChunkChoice(choice: ChatCompletionChunk.Choice, jsonMode?: boolean): CandidateData; export declare function toOpenAiRequestBody(modelName: string, request: GenerateRequest<typeof DeepSeekConfigSchema>): OpenAI.Chat.Completions.ChatCompletionCreateParamsNonStreaming; export declare function gptRunner(name: string, client: OpenAI): (request: GenerateRequest<typeof DeepSeekConfigSchema>, streamingCallback?: StreamingCallback<GenerateResponseChunkData>) => Promise<GenerateResponseData>;