Package model
class ModelConfig
func init
init(service: String, kind: String, name: String, apiKey: String = "", baseURL: String = "")
- 描述: 初始化模型配置。如果未指定apiKey,将使用XX_API_KEY;如果未指定baseURL,将使用XX_BASE_URL。
- 参数:
service: String, 服务名称
kind: String, 模型类型
name: String, 模型名称
apiKey: String, API密钥,默认为空字符串
baseURL: String, 基础URL,默认为空字符串
struct ModelManager
func createChatModel
static func createChatModel(modelName: String, temperature: Option<Float64> = None): ChatModel
- 描述: 创建聊天模型。
- 参数:
modelName: String, 模型名称
temperature: Option<Float64>, 温度参数,控制生成文本的随机性
func createChatModel
static func createChatModel(modelConfig: ModelConfig, temperature: Option<Float64> = None): ChatModel
- 描述: 根据模型配置创建聊天模型。
- 参数:
modelConfig: ModelConfig, 模型配置
temperature: Option<Float64>, 温度参数,控制生成文本的随机性
func createEmbeddingModel
static func createEmbeddingModel(modelName: String): EmbeddingModel
func createEmbeddingModel
static func createEmbeddingModel(modelConfig: ModelConfig): EmbeddingModel
- 描述: 根据模型配置创建嵌入模型。
- 参数:
modelConfig: ModelConfig, 模型配置
func createImageModel
static func createImageModel(modelName: String): ImageModel
func createImageModel
static func createImageModel(modelConfig: ModelConfig): ImageModel
- 描述: 根据模型配置创建图像模型。
- 参数:
modelConfig: ModelConfig, 模型配置
func registerChatModel
static func registerChatModel(modelName: String, buildFn: () -> ChatModel): Unit
- 描述: 注册聊天模型。
- 参数:
modelName: String, 模型名称
buildFn: () -> ChatModel, 构建聊天模型的函数
func registerEmbeddingModel
static func registerEmbeddingModel(modelName: String, buildFn: () -> EmbeddingModel): Unit
- 描述: 注册嵌入模型。
- 参数:
modelName: String, 模型名称
buildFn: () -> EmbeddingModel, 构建嵌入模型的函数
func registerImageModel
static func registerImageModel(modelName: String, buildFn: () -> ImageModel): Unit
- 描述: 注册图像模型。
- 参数:
modelName: String, 模型名称
buildFn: () -> ImageModel, 构建图像模型的函数
struct ModelUtils
func agentMakeChat
public static func agentMakeChat(agent: Agent, messages: Array<ChatMessage>): Option<ChatMessage>
- 描述: 根据给定的代理和消息列表,生成聊天消息
- 参数:
agent: Agent, 代理对象
messages: Array<ChatMessage>, 消息列表
func agentMakeChatGet
public static func agentMakeChatGet<T>(agent: Agent, messages: Array<ChatMessage>, getFn!: (ChatMessage) -> Option<T>): Option<T>
- 描述: 根据给定的代理和消息列表,生成聊天消息并应用获取函数
- 参数:
agent: Agent, 代理对象
messages: Array<ChatMessage>, 消息列表
getFn: (ChatMessage) -> Option<T>, 获取函数,用于从聊天消息中提取特定类型的数据
func makeChat
public static func makeChat(model: ChatModel, dialog: Dialog, temperature!: Option<Float64> = None, stop!: Option<Array<String>> = None): Option<ChatMessage>
- 描述: 根据给定的聊天模型和对话,生成聊天消息
- 参数:
model: ChatModel, 聊天模型
dialog: Dialog, 对话内容
temperature: Option<Float64>, 温度参数,控制生成消息的随机性
stop: Option<Array<String>>, 停止词列表,用于控制生成消息的终止条件
func makeChat
public static func makeChat(model: ChatModel, messages: Array<ChatMessage>, temperature!: Option<Float64> = None, stop!: Option<Array<String>> = None): Option<ChatMessage>
- 描述: 根据给定的聊天模型和消息列表,生成聊天消息
- 参数:
model: ChatModel, 聊天模型
messages: Array<ChatMessage>, 消息列表
temperature: Option<Float64>, 温度参数,控制生成消息的随机性
stop: Option<Array<String>>, 停止词列表,用于控制生成消息的终止条件
func makeChat
public static func makeChat(name: String, model: ChatModel, dialog: Dialog, temperature!: Option<Float64> = None, stop!: Option<Array<String>> = None): Option<ChatMessage>
- 描述: 根据给定的名称、聊天模型和对话,生成聊天消息
- 参数:
name: String, 名称标识
model: ChatModel, 聊天模型
dialog: Dialog, 对话内容
temperature: Option<Float64>, 温度参数,控制生成消息的随机性
stop: Option<Array<String>>, 停止词列表,用于控制生成消息的终止条件
func makeChat
public static func makeChat(name: String, model: ChatModel, messages: Array<ChatMessage>, temperature!: Option<Float64> = None, stop!: Option<Array<String>> = None): Option<ChatMessage>
- 描述: 根据给定的名称、聊天模型和消息列表,生成聊天消息
- 参数:
name: String, 名称标识
model: ChatModel, 聊天模型
messages: Array<ChatMessage>, 消息列表
temperature: Option<Float64>, 温度参数,控制生成消息的随机性
stop: Option<Array<String>>, 停止词列表,用于控制生成消息的终止条件
func makeChatGet
public static func makeChatGet<T>(name: String, model: ChatModel, messages: Array<ChatMessage>, getFn!: (ChatMessage) -> Option<T>): Option<T>
- 描述: 根据给定的名称、聊天模型和消息列表,生成聊天消息并应用获取函数
- 参数:
name: String, 名称标识
model: ChatModel, 聊天模型
messages: Array<ChatMessage>, 消息列表
getFn: (ChatMessage) -> Option<T>, 获取函数,用于从聊天消息中提取特定类型的数据