mtmsdk 0.0.41 → 0.0.43

package/src/adk/core/models/LiveRequest.ts

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+export declare interface LiveRequest {
+  content?: any;
+  blob?: any;
+  close?: boolean;
+  model_config?: any;
+}

package/src/adk/core/models/RuntimeConfig.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * Runtime configuration for the ADK web app. Corresponds to the JSON file
+ * read by the web app at runtime (runtime-config.json).
+ */
+export declare interface RuntimeConfig {
+  backendUrl: string;
+  logo?: LogoConfig;
+}
+
+/**
+ * Logo configuration for the ADK web app.
+ */
+export declare interface LogoConfig {
+  text: string;
+  imageUrl: string;
+}

package/src/adk/core/models/Session.ts

@@ -0,0 +1,52 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import z from "zod";
+import { type Event, zAdkEvent } from "./types";
+
+/**
+ * Interface representing the state of a Session.
+ */
+export type SessionState = {
+	[key: string]: unknown;
+};
+
+/**
+ * Interface representing a Session.
+ */
+export declare interface Session {
+	id?: string;
+	appName?: string;
+	userId?: string;
+	state?: SessionState;
+	events?: Event[];
+	lastUpdateTime?: number;
+}
+
+export const zAdkSessionState = z.any();
+
+export const zAdkSession = z.object({
+	id: z.string().optional(),
+	appName: z.string().optional(),
+	userId: z.string().optional(),
+	state: zAdkSessionState,
+	events: z.array(zAdkEvent).optional(),
+  // lastUpdateTime: z.number(), //删除
+	updateAt: z.iso.datetime().optional(), //新增
+	// updateAt: z.coerce.date(),
+	// updateAt: z.date().optional(), //新增
+});
+

package/src/adk/core/models/Trace.ts

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+export declare interface Span {
+  name: string;
+  start_time: number;
+  end_time: number;
+  span_id: string;
+  parent_span_id?: string;
+  trace_id: string;
+  attributes?: any;
+  children?: Span[];
+  invoc_id?: string;
+  // For backward compatibility.
+  'gcp.vertex.agent.llm_request'?: string;
+  'gcp.vertex.agent.llm_response'?: string;
+}
+
+export declare interface SpanNode extends Span {
+  children: SpanNode[];
+  depth: number;
+  duration: number;
+  id: string;  // Using span_id as string ID
+}
+
+export declare interface TimeTick {
+  position: number;
+  label: string;
+}

package/src/adk/core/models/types.ts

@@ -0,0 +1,255 @@
+import z from "zod";
+
+// 修复拼写错误: mimtType -> mimeType
+export const zAdkBlob = z.object({
+  mimeType: z.string().optional(),
+  data: z.string(),
+}).meta({ref:"AdkBlob"})
+
+export type Blob = z.infer<typeof zAdkBlob>
+
+export const zAdkFunctionCall = z.object({
+  id: z.string().optional(),
+  name: z.string(),
+  // 明确 args 为任意键值对
+  args: z.record(z.string(), z.any()),
+}).meta({ref: "AdkFunctionCall"})
+
+export type FunctionCall = z.infer<typeof zAdkFunctionCall>
+
+export const zAdkFunctionResp = z.object({
+  id: z.string().optional(),
+  name: z.string(),
+  response: z.record(z.string(),z.any()),
+}).meta({ref: "AdkFunctionResp"})
+
+export type FunctionResponse = z.infer<typeof zAdkFunctionResp>
+
+export const zAdkFileData = z.object({
+  mimeType: z.string(),
+  fileUri: z.string(),
+}).meta({ref: "AdkFileData"})
+
+export type FileData = z.infer<typeof zAdkFileData>
+
+export const zAdkExecutableCode = z.object({
+  language: z.enum(["UNKNOWN","PYTHON"]),
+  code: z.string(),
+}).meta({ref:"AdkExecutableCode"})
+
+export type ExecutableCode = z.infer<typeof zAdkExecutableCode>
+
+export const zAdkCodeExecutionResult = z.object({
+  outcome: z.enum(['UNKNOWN','OK','FAILED','DEADLINE_EXCEEDED']),
+  output: z.string(),
+}).meta({ref:"AdkCodeExecutionResult"})
+
+export type CodeExecutionResult = z.infer<typeof zAdkCodeExecutionResult>
+
+// Part 是递归结构的一部分，但在 Zod 中通常不需要显式递归定义，除非 Part 包含 GenAiContent
+export const zAdkPart = z.object({
+    text: z.string().optional(),
+    inlineData: zAdkBlob.optional(),
+    functionCall: zAdkFunctionCall.optional(),
+    functionResponse: zAdkFunctionResp.optional(),
+    thought: z.boolean().optional(),
+    fileData: zAdkFileData.optional(),
+    executableCode: zAdkExecutableCode.optional(),
+    codeExecutionResult: zAdkCodeExecutionResult.optional(),
+}).meta({ref:"AdkPart"})
+
+export type Part = z.infer<typeof zAdkPart>
+
+export declare interface GenAiContent {
+  role: string;
+  parts: Part[];
+}
+
+export const zGenaiContent = z.object({
+  role: z.string(),
+  parts: z.array(zAdkPart),
+}).meta({ref:"GenaiContent"})
+
+export declare interface LlmRequest {
+  contents: GenAiContent[];
+}
+
+export const zAdkLlmRequest = z.object({
+  contents: z.array(zGenaiContent)
+}).meta({ref:"AdkLlmRequest"})
+
+export const zAdkEventActions = z.object({
+  message: z.string().optional(),
+  artifactDelta: z.any().optional(),
+  functionCall: zAdkFunctionCall.optional(),
+  functionResponse: zAdkFunctionResp.optional(),
+  finishReason: z.string().optional(),
+}).meta({ref:"AdkEventActions"})
+
+export type EventActions = z.infer<typeof zAdkEventActions>
+
+// 注意: 为了避免类型混淆和 'llmResponse' 不存在的错误，
+// 这里我们将 Event 定义为扁平结构，直接包含 Content 和 Metadata。
+// 这也是 Google ADK Event Python 对象的实际结构。
+export const zAdkEvent = z.object({
+  // 基础字段
+  id: z.string().optional(),
+  author: z.string().optional(),
+  invocationId: z.string().optional(),
+  actions: zAdkEventActions.optional(),
+  longRunningToolIds: z.array(z.string()).optional(),
+  branch: z.string().optional(),
+  timestamp: z.number().optional(), // 对应 Unix Timestamp (seconds or millis)
+
+  // LLM Response 字段 (扁平化)
+  content: zGenaiContent.optional(),
+  error: z.string().optional(),
+  errorMessage: z.string().optional(),
+  errorCode: z.string().optional(),
+
+  // Metadata
+  groundingMetadata: z.any().optional(), // 暂时使用 any，如有具体结构可细化
+  usageMetadata: z.any().optional(),
+  citationMetadata: z.any().optional(),
+  customMetadata: z.any().optional(),
+
+  // 状态
+  turnComplete: z.boolean().optional(),
+  interrupted: z.boolean().optional(),
+}).meta({ref:"AdkEvent"})
+
+export type Event = z.infer<typeof zAdkEvent>
+
+/**
+ * 数据库对应的 Event 结构，通常与 API Event 结构一致
+ */
+export const zDbAdkEvent = zAdkEvent.meta({ref:"DbAdkEvent"})
+// import z from "zod";
+
+// export const zAdkBlob = z.object({
+//   mimtType: z.string().optional(),
+//   data: z.string(),
+// }).meta({ref:"AdkBlob"})
+
+// export type Blob = z.infer<typeof zAdkBlob>
+
+// export const zAdkFunctionCall = z.object({
+//   id: z.string().optional(),
+//   name: z.string(),
+//   args: z.record(z.string(), z.any()), // 最终的类型应该是: {[key: string]: any}
+// }).meta({ref: "AdkFunctionCall"})
+// export type FunctionCall = z.infer<typeof zAdkFunctionCall>
+
+// export const zAdkFunctionResp =z.object({
+//   id: z.string().optional(),
+//   name: z.string(),
+//   response: z.record(z.string(),z.any()),
+// }).meta({ref: "AdkFunctionResp"})
+
+// export type FunctionResponse = z.infer<typeof zAdkFunctionResp>
+
+// export const zAdkFileData = z.object({
+//   mimeType: z.string(),
+//   fileUri: z.string(),
+// }).meta({ref: "AdkFileData"})
+
+// export type FileData = z.infer<typeof zAdkFileData>
+
+// export const zAdkExecutableCode = z.object({
+//   language: z.enum(["UNKNOWN","PYTHON"]),
+//   code: z.string(),
+// }).meta({ref:"AdkExecutableCode"})
+
+// export type  ExecutableCode= z.infer<typeof zAdkExecutableCode>
+
+// export const zAdkCodeExecutionResult = z.object({
+//   outcome: z.enum(['UNKNOWN','OK','FAILED','DEADLINE_EXCEEDED']),
+//   output:z.string(),
+// }).meta({ref:"AdkCodeExecutionResult"})
+
+// export type CodeExecutionResult = z.infer<typeof zAdkCodeExecutionResult>
+
+// export const zAdkPart = z.object(
+//   {
+//     text: z.string().optional(),
+//     inlineData: zAdkBlob.optional(),
+//     functionCall: zAdkFunctionCall.optional(),
+//     functionResponse: zAdkFunctionResp.optional(),
+//     thought: z.boolean().optional(),
+//     fileData: zAdkFileData.optional(),
+//     executableCode:zAdkExecutableCode.optional(),
+//     codeExecutionResult: zAdkCodeExecutionResult.optional(),
+//   }
+// ).meta({ref:"AdkPart"})
+
+// export type Part = z.infer<typeof zAdkPart>
+
+// export declare interface GenAiContent {
+//   role: string;
+//   parts: Part[];
+// }
+// export const zGenaiContent = z.object({
+//   role: z.string(),
+//   parts:z.array(zAdkPart),
+// }).meta({ref:"GenaiContent"})
+
+
+// export declare interface LlmRequest {
+//   contents: GenAiContent[];
+// }
+
+// export const zAdkLlmRequest = z.object({
+//   contents: z.array(zGenaiContent)
+// }).meta({ref:"AdkLlmRequest"})
+
+// export declare interface LlmResponse {
+//   content: GenAiContent;
+//   error?: string;
+//   errorMessage?: string;
+//   errorCode?: string;
+//   longRunningToolIds?: string[];
+// }
+
+// export const zLLmResponse = z.object({
+//   content: zGenaiContent,
+//   error:z.string().optional(),
+//   errorMessage: z.string().optional(),
+//   errorCode: z.string().optional(),
+//   longRunningToolIds: z.array(z.string()).optional()
+// }).meta({ref:"LLmResponse"})
+
+// export const zAdkEventActions = z.object({
+//   message: z.string().optional(),
+//   artifactDelta: z.any().optional(),
+//   functionCall: zAdkFunctionCall.optional(),
+//   functionResponse:zAdkFunctionResp.optional(),
+//   finishReason:z.string().optional(),
+// }).meta({ref:"AdkEventActions"})
+
+// export type EventActions = z.infer<typeof zAdkEventActions>
+
+// export const zAdkEvent = zLLmResponse.extend({
+//   id: z.string().optional(),
+//   author: z.string().optional(),
+//   invocationId: z.string().optional(),
+//   actions: zAdkEventActions.optional(),
+//   longRunningToolIds: z.array(z.string()).optional(),
+//   branch:z.string().optional(),
+//   timestamp:z.number().optional(),
+// }).meta({ref:"AdkEvent"})
+
+// export type Event = z.infer<typeof zAdkEvent>
+
+// /**
+//  * 跟数据库结构对应
+//  */
+// export const zDbAdkEvent = zLLmResponse.extend({
+//   id: z.string().optional(),
+//   author: z.string().optional(),
+//   invocationId: z.string().optional(),
+//   actions: zAdkEventActions.optional(),
+//   longRunningToolIds: z.array(z.string()).optional(),
+//   branch:z.string().optional(),
+//   timestamp:z.number().optional(),
+// }).meta({ref:"DbAdkEvent"})
+