chat(@RequestBody ChatRequest request) {
+ ChatCompletionCreateParams params = ChatCompletionCreateParams.builder()
+ .model(Constants.ModelChatGLM4)
+ .messages(Arrays.asList(
+ ChatMessage.builder()
+ .role(ChatMessageRole.USER.value())
+ .content(request.getMessage())
+ .build()
+ ))
+ .build();
+
+ ChatCompletionResponse response = zaiClient.chat().createChatCompletion(params);
+
+ if (response.isSuccess()) {
+ String content = response.getData().getChoices().get(0).getMessage().getContent();
+ return ResponseEntity.ok(content);
+ } else {
+ return ResponseEntity.badRequest().body(response.getMsg());
+ }
+ }
+}
+```
+
+## 🔧 可用服务
+
+### Chat 服务
+- 对话补全（同步/异步）
+- 流式对话
+- 函数调用
+- 多模态对话（文本+图像）
+
+### Embeddings 服务
+- 文本向量化
+- 批量嵌入
+- 多种嵌入模型
+
+### Images 服务
+- 文本到图像生成
+- 图像编辑
+- 图像变体
+
+### Audio 服务
+- 语音转文本
+- 文本转语音
+- 语音翻译
+
+### Files 服务
+- 文件上传
+- 文件管理
+- 文件检索
+
+### Assistants 服务
+- AI助手创建
+- 助手管理
+- 对话线程
+
+### Agents 服务
+- 智能代理
+- 工作流管理
+- 任务执行
+
+### Knowledge 服务
+- 知识库管理
+- 文档处理
+- 知识检索
+
+### Batch 服务
+- 批量处理
+- 异步任务
+- 结果管理
+
+## 🤖 支持的模型
+
+### 文本生成模型
+- `glm-4-plus` - 最新的GLM-4 Plus模型
+- `glm-4-0520` - GLM-4标准版
+- `glm-4-long` - 长文本处理版本
+- `glm-4-airx` - 轻量级版本
+- `glm-4-air` - 快速响应版本
+- `glm-4-flashx` - 超快响应版本
+- `glm-4-flash` - 闪电版本
+
+### 视觉模型
+- `glm-4v-plus` - 多模态理解模型
+- `glm-4v` - 视觉理解模型
+
+### 图像生成模型
+- `cogview-3-plus` - 高质量图像生成
+- `cogview-3` - 标准图像生成
+
+### 嵌入模型
+- `embedding-3` - 最新嵌入模型
+- `embedding-2` - 标准嵌入模型
+
+### 专业模型
+- `charglm-3` - 角色扮演模型
+- `emohaa` - 情感分析模型
+
+## 📈 版本更新
+
+详细的版本更新记录和历史信息，请查看 [Release-Note.md](Release-Note.md)。
+
+## 📄 许可证
+
+本项目基于 MIT 许可证开源 - 详情请查看 [LICENSE](LICENSE) 文件。
+
+## 🤝 贡献
+
+欢迎贡献代码！请随时提交 Pull Request。
+
+## 📞 支持
+
+如有问题和技术支持:
+- Visit [Z.ai Platform](https://z.ai/)
+- Visit [ZHIPU AI Open Platform](http://open.bigmodel.cn/)
+- Check our [Architecture Documentation](ARCHITECTURE.md)
diff --git a/Release-Note.md b/Release-Note.md
new file mode 100644
index 0000000..f1df9cf
--- /dev/null
+++ b/Release-Note.md
@@ -0,0 +1,4 @@
+# Release Notes
+
+### 0.0.1
+- ✅ Initial the Z.ai sdk refer from ZHIPU sdk
\ No newline at end of file
diff --git a/core/pom.xml b/core/pom.xml
new file mode 100644
index 0000000..aa36953
--- /dev/null
+++ b/core/pom.xml
@@ -0,0 +1,237 @@
+
+
+ 4.0.0
+
+ ai.z.openapi
+ z-ai-sdk-parent
+ ${revision}
+ ../pom.xml
+
+ z-ai-sdk
+
+ jar
+
+ Java sdk for Z.ai open api
+ Java SDK for Z.ai Open Platform API
+ https://z.ai
+
+
+ MIT License
+ https://www.opensource.org/licenses/mit-license.php
+ repo
+
+
+
+
+ Z.ai
+ Z.ai
+ user_feedback@z.ai
+ https://z.ai/model-api
+ Z.ai
+ https://z.ai
+
+ architect
+ developer
+
+ Asia/Shanghai
+
+
+
+ scm:git:https://github.com/THUDM/z-ai-sdk-java.git
+ scm:git:https://github.com/THUDM/z-ai-sdk-java.git
+ HEAD
+ https://github.com/THUDM/z-ai-sdk-java
+
+
+
+ 8
+ 8
+ UTF-8
+ 2.0.11
+ 3.14.9
+ 2.11.3
+ 2.9.0
+ 3.1.8
+ 4.2.2
+ 1.18.32
+ 2.9.0
+ 5.10.2
+ 1.19.8
+ 5.15.0
+
+
+
+
+
+ org.slf4j
+ slf4j-api
+ ${slf4j.version}
+
+
+ org.slf4j
+ slf4j-simple
+ ${slf4j.version}
+ test
+
+
+ org.slf4j
+ slf4j-log4j12
+ ${slf4j.version}
+ pom
+ test
+
+
+
+
+ com.squareup.okhttp3
+ okhttp-sse
+ ${okhttp.version}
+
+
+ com.squareup.okhttp3
+ okhttp
+ ${okhttp.version}
+
+
+ com.squareup.okhttp3
+ logging-interceptor
+ ${okhttp.version}
+
+
+ io.reactivex.rxjava3
+ rxjava
+ ${rxjava.version}
+
+
+ com.squareup.retrofit2
+ adapter-rxjava2
+ ${retrofit2.version}
+
+
+ com.squareup.retrofit2
+ converter-jackson
+ ${retrofit2.version}
+
+
+
+
+ com.auth0
+ java-jwt
+ ${jwt.version}
+
+
+ com.fasterxml.jackson.core
+ jackson-annotations
+
+
+ com.fasterxml.jackson.core
+ jackson-databind
+
+
+ com.fasterxml.jackson.core
+ jackson-core
+
+
+
+
+
+
+ com.fasterxml.jackson.core
+ jackson-databind
+ ${jackson.version}
+
+
+ com.fasterxml.jackson.core
+ jackson-core
+ ${jackson.version}
+
+
+ com.fasterxml.jackson.core
+ jackson-annotations
+ ${jackson.version}
+
+
+
+
+ org.projectlombok
+ lombok
+ ${lombok.version}
+
+
+
+
+ org.apache.tika
+ tika-core
+ ${tika.version}
+
+
+
+
+ org.junit.jupiter
+ junit-jupiter-api
+ ${junit.test.version}
+ test
+
+
+ org.junit.jupiter
+ junit-jupiter-params
+ ${junit.test.version}
+ test
+
+
+ org.junit.jupiter
+ junit-jupiter-engine
+ ${junit.test.version}
+ test
+
+
+
+
+ org.testcontainers
+ testcontainers
+ ${testcontainer.version}
+ test
+
+
+ org.testcontainers
+ mockserver
+ ${testcontainer.version}
+ test
+
+
+ org.testcontainers
+ junit-jupiter
+ ${testcontainer.version}
+ test
+
+
+
+
+ org.mock-server
+ mockserver-netty
+ ${mockserver.version}
+ test
+
+
+ org.mock-server
+ mockserver-client-java
+ ${mockserver.version}
+ test
+
+
+
+
+
+
+ org.apache.maven.plugins
+ maven-compiler-plugin
+ 3.8.1
+
+
+ org.apache.maven.plugins
+ maven-surefire-plugin
+ 2.22.2
+
+
+
+
diff --git a/core/src/main/java/ai/z/openapi/ZaiClient.java b/core/src/main/java/ai/z/openapi/ZaiClient.java
new file mode 100644
index 0000000..abe9ccc
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/ZaiClient.java
@@ -0,0 +1,646 @@
+package ai.z.openapi;
+
+import ai.z.openapi.service.AbstractClientBaseService;
+import ai.z.openapi.service.model.ChatError;
+import ai.z.openapi.service.model.ZAiHttpException;
+import ai.z.openapi.service.chat.ChatService;
+import ai.z.openapi.service.chat.ChatServiceImpl;
+import ai.z.openapi.service.agents.AgentService;
+import ai.z.openapi.service.agents.AgentServiceImpl;
+import ai.z.openapi.service.embedding.EmbeddingService;
+import ai.z.openapi.service.embedding.EmbeddingServiceImpl;
+import ai.z.openapi.service.file.FileService;
+import ai.z.openapi.service.file.FileServiceImpl;
+import ai.z.openapi.service.audio.AudioService;
+import ai.z.openapi.service.audio.AudioServiceImpl;
+import ai.z.openapi.service.image.ImageService;
+import ai.z.openapi.service.image.ImageServiceImpl;
+import ai.z.openapi.service.batches.BatchService;
+import ai.z.openapi.service.batches.BatchServiceImpl;
+import ai.z.openapi.service.fine_turning.FineTuningService;
+import ai.z.openapi.service.fine_turning.FineTuningServiceImpl;
+import ai.z.openapi.service.web_search.WebSearchService;
+import ai.z.openapi.service.web_search.WebSearchServiceImpl;
+import ai.z.openapi.service.videos.VideosService;
+import ai.z.openapi.service.videos.VideosServiceImpl;
+import ai.z.openapi.service.knowledge.KnowledgeService;
+import ai.z.openapi.service.knowledge.KnowledgeServiceImpl;
+import ai.z.openapi.service.document.DocumentService;
+import ai.z.openapi.service.document.DocumentServiceImpl;
+import ai.z.openapi.service.assistant.AssistantService;
+import ai.z.openapi.service.assistant.AssistantServiceImpl;
+import ai.z.openapi.core.config.ZaiConfig;
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.core.model.FlowableClientResponse;
+import ai.z.openapi.utils.FlowableRequestSupplier;
+import ai.z.openapi.utils.OkHttps;
+import ai.z.openapi.utils.RequestSupplier;
+import ai.z.openapi.utils.StringUtils;
+import io.reactivex.Flowable;
+import io.reactivex.Single;
+import okhttp3.OkHttpClient;
+import okhttp3.ResponseBody;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import retrofit2.Retrofit;
+import retrofit2.adapter.rxjava2.RxJava2CallAdapterFactory;
+import retrofit2.converter.jackson.JacksonConverterFactory;
+
+import java.lang.reflect.InvocationTargetException;
+import java.util.concurrent.TimeUnit;
+
+import static ai.z.openapi.core.Constants.ZHIPU_AI_BASE_URL;
+import static ai.z.openapi.core.Constants.Z_AI_BASE_URL;
+
+/**
+ * ZaiClient is the main entry point for interacting with the Z.ai API. This client
+ * provides access to various AI services including chat, embeddings, file operations,
+ * audio processing, image generation, and more.
+ *
+ *
+ * The client supports both Z.ai and ZHIPU AI endpoints and provides thread-safe lazy
+ * initialization of services.
+ *
+ *
+ *
+ * Example usage:
+ *
+ * {@code
+ * ZaiClient client = new ZaiClient.Builder("your-api-key")
+ * .networkConfig(30, 10, 30, 30, TimeUnit.SECONDS)
+ * .build();
+ *
+ * ChatService chatService = client.chat();
+ * // Use the chat service...
+ *
+ * client.close(); // Don't forget to close when done
+ * }
+ *
+ */
+public class ZaiClient extends AbstractClientBaseService {
+
+ /** Logger instance for this class */
+ private static final Logger logger = LoggerFactory.getLogger(ZaiClient.class);
+
+ /** HTTP client for making network requests */
+ private final OkHttpClient httpClient;
+
+ /** Retrofit instance for API communication */
+ private final Retrofit retrofit;
+
+ // Service instances - lazily initialized for thread safety and performance
+ /** Chat service for conversational AI operations */
+ private ChatService chatService;
+
+ /** Agent service for AI agent management */
+ private AgentService agentService;
+
+ /** Embedding service for text embeddings */
+ private EmbeddingService embeddingService;
+
+ /** File service for file operations */
+ private FileService fileService;
+
+ /** Audio service for audio processing */
+ private AudioService audioService;
+
+ /** Image service for image generation and processing */
+ private ImageService imageService;
+
+ /** Batch service for batch processing operations */
+ private BatchService batchService;
+
+ /** Fine-tuning service for model customization */
+ private FineTuningService fineTuningService;
+
+ /** Web search service for internet search capabilities */
+ private WebSearchService webSearchService;
+
+ /** Videos service for video processing */
+ private VideosService videosService;
+
+ /** Knowledge service for knowledge base operations */
+ private KnowledgeService knowledgeService;
+
+ /** Document service for document processing */
+ private DocumentService documentService;
+
+ /** Assistant service for AI assistant functionality */
+ private AssistantService assistantService;
+
+ /**
+ * Constructs a new ZaiClient with the specified configuration. By default, this
+ * client uses the Z.ai OpenAPI endpoint.
+ * @param config the configuration object containing API keys, timeouts, and other
+ * settings
+ * @throws IllegalArgumentException if config is null or invalid
+ */
+ public ZaiClient(ZaiConfig config) {
+ this.httpClient = OkHttps.create(config);
+ this.retrofit = new Retrofit.Builder()
+ .baseUrl(StringUtils.isEmpty(config.getBaseUrl()) ? Z_AI_BASE_URL : config.getBaseUrl())
+ .client(httpClient)
+ .addConverterFactory(JacksonConverterFactory.create(mapper))
+ .addCallAdapterFactory(RxJava2CallAdapterFactory.create())
+ .build();
+ }
+
+ // ==================== Service Accessor Methods ====================
+ // All service methods use lazy initialization with synchronization for thread safety
+
+ /**
+ * Returns the chat service for conversational AI operations. This service handles
+ * chat completions, streaming conversations, and related functionality.
+ * @return the ChatService instance (lazily initialized)
+ */
+ public synchronized ChatService chat() {
+ if (chatService == null) {
+ this.chatService = new ChatServiceImpl(this);
+ }
+ return chatService;
+ }
+
+ /**
+ * Returns the agent service for AI agent management. This service handles agent
+ * creation, configuration, and execution.
+ * @return the AgentService instance (lazily initialized)
+ */
+ public synchronized AgentService agents() {
+ if (agentService == null) {
+ this.agentService = new AgentServiceImpl(this);
+ }
+ return agentService;
+ }
+
+ /**
+ * Returns the embedding service for text embeddings. This service converts text into
+ * numerical vector representations.
+ * @return the EmbeddingService instance (lazily initialized)
+ */
+ public synchronized EmbeddingService embeddings() {
+ if (embeddingService == null) {
+ this.embeddingService = new EmbeddingServiceImpl(this);
+ }
+ return embeddingService;
+ }
+
+ /**
+ * Returns the file service for file operations. This service handles file uploads,
+ * downloads, and management.
+ * @return the FileService instance (lazily initialized)
+ */
+ public synchronized FileService files() {
+ if (fileService == null) {
+ this.fileService = new FileServiceImpl(this);
+ }
+ return fileService;
+ }
+
+ /**
+ * Returns the audio service for audio processing. This service handles
+ * speech-to-text, text-to-speech, and audio analysis.
+ * @return the AudioService instance (lazily initialized)
+ */
+ public synchronized AudioService audio() {
+ if (audioService == null) {
+ this.audioService = new AudioServiceImpl(this);
+ }
+ return audioService;
+ }
+
+ /**
+ * Returns the image service for image generation and processing. This service handles
+ * image creation, editing, and analysis.
+ * @return the ImageService instance (lazily initialized)
+ */
+ public synchronized ImageService images() {
+ if (imageService == null) {
+ this.imageService = new ImageServiceImpl(this);
+ }
+ return imageService;
+ }
+
+ /**
+ * Returns the batch service for batch processing operations. This service handles
+ * large-scale batch processing of requests.
+ * @return the BatchService instance (lazily initialized)
+ */
+ public synchronized BatchService batches() {
+ if (batchService == null) {
+ this.batchService = new BatchServiceImpl(this);
+ }
+ return batchService;
+ }
+
+ /**
+ * Returns the fine-tuning service for model customization. This service handles
+ * training custom models on user data.
+ * @return the FineTuningService instance (lazily initialized)
+ */
+ public synchronized FineTuningService fineTuning() {
+ if (fineTuningService == null) {
+ this.fineTuningService = new FineTuningServiceImpl(this);
+ }
+ return fineTuningService;
+ }
+
+ /**
+ * Returns the web search service for internet search capabilities. This service
+ * provides AI-powered web search functionality.
+ * @return the WebSearchService instance (lazily initialized)
+ */
+ public synchronized WebSearchService webSearch() {
+ if (webSearchService == null) {
+ this.webSearchService = new WebSearchServiceImpl(this);
+ }
+ return webSearchService;
+ }
+
+ /**
+ * Returns the videos service for video processing. This service handles video
+ * analysis, generation, and manipulation.
+ * @return the VideosService instance (lazily initialized)
+ */
+ public synchronized VideosService videos() {
+ if (videosService == null) {
+ this.videosService = new VideosServiceImpl(this);
+ }
+ return videosService;
+ }
+
+ /**
+ * Returns the knowledge service for knowledge base operations. This service manages
+ * knowledge bases and retrieval operations.
+ * @return the KnowledgeService instance (lazily initialized)
+ */
+ public synchronized KnowledgeService knowledge() {
+ if (knowledgeService == null) {
+ this.knowledgeService = new KnowledgeServiceImpl(this);
+ }
+ return knowledgeService;
+ }
+
+ /**
+ * Returns the document service for document processing. This service handles document
+ * parsing, analysis, and manipulation.
+ * @return the DocumentService instance (lazily initialized)
+ */
+ public synchronized DocumentService documents() {
+ if (documentService == null) {
+ this.documentService = new DocumentServiceImpl(this);
+ }
+ return documentService;
+ }
+
+ /**
+ * Returns the assistant service for AI assistant functionality. This service provides
+ * advanced AI assistant capabilities.
+ * @return the AssistantService instance (lazily initialized)
+ */
+ public synchronized AssistantService assistants() {
+ if (assistantService == null) {
+ this.assistantService = new AssistantServiceImpl(this);
+ }
+ return assistantService;
+ }
+
+ // ==================== Utility Methods ====================
+
+ /**
+ * Returns the underlying Retrofit instance used for API communication. This method is
+ * primarily intended for advanced users who need direct access to the Retrofit client
+ * for custom API calls.
+ * @return the Retrofit instance
+ */
+ public Retrofit retrofit() {
+ return retrofit;
+ }
+
+ /**
+ * Closes the ZAi client and releases all associated resources. This method shuts down
+ * the HTTP client's connection pool and executor service. After calling this method,
+ * the client should not be used for further requests.
+ *
+ *
+ * Important: Always call this method when you're done with the
+ * client to prevent resource leaks.
+ *
+ */
+ public void close() {
+ if (httpClient != null) {
+ httpClient.dispatcher().executorService().shutdown();
+ }
+ }
+
+ // ==================== Core Request Execution Methods ====================
+
+ /**
+ * Executes a synchronous API request and returns the response. This method handles
+ * the complete request lifecycle including error handling and response wrapping.
+ * @param the type of data expected in the response
+ * @param the type of parameters for the request
+ * @param the type of client request
+ * @param the type of client response
+ * @param request the client request containing parameters
+ * @param requestSupplier the supplier that creates the actual API call
+ * @param tRespClass the class of the response type
+ * @return the wrapped response containing either success data or error information
+ */
+ @Override
+ public , TResp extends ClientResponse> TResp executeRequest(
+ TReq request, RequestSupplier requestSupplier, Class tRespClass) {
+ Single apiCall = requestSupplier.get((Param) request);
+
+ TResp tResp = convertToClientResponse(tRespClass);
+ try {
+ // Execute the API call synchronously
+ Data response = execute(apiCall);
+ tResp.setCode(200);
+ tResp.setMsg("Call successful");
+ tResp.setData(response);
+ tResp.setSuccess(true);
+ }
+ catch (ZAiHttpException e) {
+ logger.error("API request failed with business error", e);
+ tResp.setCode(e.statusCode);
+ tResp.setMsg("Business error");
+ tResp.setSuccess(false);
+ ChatError chatError = new ChatError();
+ chatError.setCode(Integer.parseInt(e.code));
+ chatError.setMessage(e.getMessage());
+ tResp.setError(chatError);
+ }
+ return tResp;
+ }
+
+ /**
+ * Executes a streaming API request and returns a response containing a Flowable
+ * stream. This method is used for requests that return data as a continuous stream,
+ * such as chat completions with streaming enabled.
+ * @param the type of data expected in each stream element
+ * @param the type of parameters for the request
+ * @param the type of client request
+ * @param the type of flowable client response
+ * @param request the client request containing parameters
+ * @param requestSupplier the supplier that creates the actual streaming API call
+ * @param tRespClass the class of the response type
+ * @param tDataClass the class of the data type for stream elements
+ * @return the wrapped response containing either a success stream or error
+ * information
+ */
+ @Override
+ @SuppressWarnings("unchecked")
+ public , TResp extends FlowableClientResponse> TResp streamRequest(
+ TReq request, FlowableRequestSupplier> requestSupplier,
+ Class tRespClass, Class tDataClass) {
+ retrofit2.Call apiCall = requestSupplier.get((Param) request);
+
+ TResp tResp = convertToClientResponse(tRespClass);
+ try {
+ // Create a streaming response using the provided API call
+ Flowable stream = stream(apiCall, tDataClass);
+ tResp.setCode(200);
+ tResp.setMsg("Stream initialized successfully");
+ tResp.setSuccess(true);
+ tResp.setFlowable(stream);
+ }
+ catch (ZAiHttpException e) {
+ logger.error("Streaming API request failed with business error", e);
+ tResp.setCode(e.statusCode);
+ tResp.setMsg("Business error");
+ tResp.setSuccess(false);
+ ChatError chatError = new ChatError();
+ chatError.setCode(Integer.parseInt(e.code));
+ chatError.setMessage(e.getMessage());
+ tResp.setError(chatError);
+ }
+ return tResp;
+ }
+
+ /**
+ * Creates a new instance of the specified response class using reflection. This
+ * helper method is used to instantiate response objects dynamically.
+ * @param the type of data contained in the response
+ * @param the type of client response
+ * @param tRespClass the class of the response type to instantiate
+ * @return a new instance of the response class
+ * @throws RuntimeException if the response object cannot be created
+ */
+ private > TResp convertToClientResponse(Class tRespClass) {
+ try {
+ return tRespClass.getDeclaredConstructor().newInstance();
+ }
+ catch (InstantiationException | IllegalAccessException | NoSuchMethodException | InvocationTargetException e) {
+ throw new RuntimeException("Failed to create response object of type: " + tRespClass.getSimpleName(), e);
+ }
+ }
+
+ // ==================== Builder Pattern Implementation ====================
+
+ /**
+ * Builder class for creating ZaiClient instances with custom configurations. This
+ * builder provides a fluent API for setting up the client with various options
+ * including authentication, network settings, and connection pooling.
+ *
+ *
+ * Example usage:
+ *
+ * {@code
+ * ZaiClient client = new ZaiClient.Builder("your-api-key")
+ * .networkConfig(30, 10, 30, 30, TimeUnit.SECONDS)
+ * .connectionPool(10, 5, TimeUnit.MINUTES)
+ * .enableTokenCache()
+ * .build();
+ * }
+ */
+ public static final class Builder {
+
+ /** Configuration object that accumulates all builder settings */
+ private final ZaiConfig config = new ZaiConfig();
+
+ public Builder() {
+ }
+
+ /**
+ * Creates a new builder with the specified API key.
+ * @param apiKey the API key for authentication
+ * @throws IllegalArgumentException if apiKey is null or empty
+ */
+ public Builder(String apiKey) {
+ if (apiKey == null || apiKey.trim().isEmpty()) {
+ throw new IllegalArgumentException("API key cannot be null or empty");
+ }
+ config.setApiKey(apiKey);
+ }
+
+ /**
+ * Creates a new builder with a custom base URL and API secret key.
+ * @param baseUrl the custom base URL for the API endpoint
+ * @param apiKey the API secret key for authentication
+ * @throws IllegalArgumentException if any parameter is null or empty
+ */
+ public Builder(String baseUrl, String apiKey) {
+ if (baseUrl == null || baseUrl.trim().isEmpty()) {
+ throw new IllegalArgumentException("Base URL cannot be null or empty");
+ }
+ if (apiKey == null || apiKey.trim().isEmpty()) {
+ throw new IllegalArgumentException("API secret key cannot be null or empty");
+ }
+ config.setBaseUrl(baseUrl);
+ config.setApiKey(apiKey);
+ }
+
+ /**
+ * Config the api service base url
+ * @param baseUrl base url
+ * @return this Builder instance for method chaining
+ */
+ public Builder baseUrl(String baseUrl) {
+ if (baseUrl == null || baseUrl.trim().isEmpty()) {
+ throw new IllegalArgumentException("Base URL cannot be null or empty");
+ }
+ config.setBaseUrl(baseUrl);
+ return this;
+ }
+
+ /**
+ * Config the apikey
+ * @param apiKey api key
+ * @return this Builder instance for method chaining
+ */
+ public Builder apiKey(String apiKey) {
+ if (apiKey == null || apiKey.trim().isEmpty()) {
+ throw new IllegalArgumentException("API secret key cannot be null or empty");
+ }
+ config.setApiKey(apiKey);
+ return this;
+ }
+
+ /**
+ * Use the ZHIPU AI base url
+ * @return this Builder instance for method chaining
+ */
+ public Builder ofZHIPU() {
+ config.setBaseUrl(ZHIPU_AI_BASE_URL);
+ return this;
+ }
+
+ /**
+ * Use the Z AI base url
+ * @return this Builder instance for method chaining
+ */
+ public Builder ofZAI() {
+ config.setBaseUrl(Z_AI_BASE_URL);
+ return this;
+ }
+
+ /**
+ * Disables token caching, forcing the client to use API keys for direct requests.
+ * This is the default behavior.
+ * @return this Builder instance for method chaining
+ */
+ public Builder disableTokenCache() {
+ config.setDisableTokenCache(true);
+ return this;
+ }
+
+ /**
+ * Enables token caching, allowing the client to use access tokens for requests.
+ * This can improve performance by reducing authentication overhead.
+ * @return this Builder instance for method chaining
+ */
+ public Builder enableTokenCache() {
+ config.setDisableTokenCache(false);
+ return this;
+ }
+
+ /**
+ * Configures the HTTP connection pool settings.
+ * @param maxIdleConnections maximum number of idle connections to keep in the
+ * pool
+ * @param keepAliveDuration how long to keep idle connections alive
+ * @param timeUnit the time unit for the keep alive duration
+ * @return this Builder instance for method chaining
+ */
+ public Builder connectionPool(int maxIdleConnections, long keepAliveDuration, TimeUnit timeUnit) {
+ config.setConnectionPoolMaxIdleConnections(maxIdleConnections);
+ config.setConnectionPoolKeepAliveDuration(keepAliveDuration);
+ config.setConnectionPoolTimeUnit(timeUnit);
+ return this;
+ }
+
+ /**
+ * Sets the token expiration time in milliseconds.
+ * @param expireMillis the token expiration time in milliseconds
+ * @return this Builder instance for method chaining
+ */
+ public Builder tokenExpire(int expireMillis) {
+ config.setExpireMillis(expireMillis);
+ return this;
+ }
+
+ /**
+ * Configures network request timeout settings.
+ * @param requestTimeOut the overall request timeout (see
+ * {@link OkHttpClient.Builder#callTimeout(long, TimeUnit)})
+ * @param connectTimeout the connection timeout (see
+ * {@link OkHttpClient.Builder#connectTimeout(long, TimeUnit)})
+ * @param readTimeout the read timeout (see
+ * {@link OkHttpClient.Builder#readTimeout(long, TimeUnit)})
+ * @param writeTimeout the write timeout (see
+ * {@link OkHttpClient.Builder#writeTimeout(long, TimeUnit)})
+ * @param timeUnit the time unit for all timeout values
+ * @return this Builder instance for method chaining
+ */
+ public Builder networkConfig(int requestTimeOut, int connectTimeout, int readTimeout, int writeTimeout,
+ TimeUnit timeUnit) {
+ config.setRequestTimeOut(requestTimeOut);
+ config.setConnectTimeout(connectTimeout);
+ config.setReadTimeout(readTimeout);
+ config.setWriteTimeout(writeTimeout);
+ config.setTimeOutTimeUnit(timeUnit);
+ return this;
+ }
+
+ /**
+ * Builds and returns a new ZaiClient instance with the configured settings.
+ * @return a new ZaiClient instance
+ * @throws IllegalStateException if the configuration is invalid
+ */
+ public ZaiClient build() {
+ return new ZaiClient(config);
+ }
+
+ }
+
+ // ==================== Static Factory Methods ====================
+
+ /**
+ * Creates a new Builder instance for constructing ZaiClient.
+ * @return a new Builder instance
+ */
+ public static Builder builder() {
+ return new Builder();
+ }
+
+ /**
+ * Creates a ZaiClient configured for ZHIPU AI platform with the specified API key.
+ * @param apiKey the API key for authentication
+ * @return a new ZaiClient instance configured for ZHIPU AI
+ * @throws IllegalArgumentException if apiKey is null or empty
+ */
+ public static Builder ofZHIPU(String apiKey) {
+ return new Builder().apiKey(apiKey).ofZHIPU();
+ }
+
+ /**
+ * Creates a ZaiClient configured for ZHIPU AI platform with the specified API key.
+ * @return a new ZaiClient instance configured for ZHIPU AI
+ * @throws IllegalArgumentException if apiKey is null or empty
+ */
+ public static Builder ofZHIPU() {
+ return new Builder().ofZHIPU();
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/agents/AgentsApi.java b/core/src/main/java/ai/z/openapi/api/agents/AgentsApi.java
new file mode 100644
index 0000000..efe0dcf
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/agents/AgentsApi.java
@@ -0,0 +1,25 @@
+package ai.z.openapi.api.agents;
+
+import ai.z.openapi.service.agents.AgentAsyncResultRetrieveParams;
+import ai.z.openapi.service.agents.AgentsCompletionRequest;
+import ai.z.openapi.service.model.ModelData;
+import io.reactivex.Single;
+import okhttp3.ResponseBody;
+import retrofit2.Call;
+import retrofit2.http.Body;
+import retrofit2.http.POST;
+import retrofit2.http.Streaming;
+
+public interface AgentsApi {
+
+ @Streaming
+ @POST("v1/agents")
+ Call agentsCompletionStream(@Body AgentsCompletionRequest request);
+
+ @POST("v1/agents")
+ Single agentsCompletionSync(@Body AgentsCompletionRequest request);
+
+ @POST("v1/agents/async-result")
+ Single queryAgentsAsyncResult(@Body AgentAsyncResultRetrieveParams request);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/assistant/AssistantApi.java b/core/src/main/java/ai/z/openapi/api/assistant/AssistantApi.java
new file mode 100644
index 0000000..2690cba
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/assistant/AssistantApi.java
@@ -0,0 +1,31 @@
+package ai.z.openapi.api.assistant;
+
+import ai.z.openapi.service.assistant.AssistantCompletion;
+import ai.z.openapi.service.assistant.AssistantParameters;
+import ai.z.openapi.service.assistant.conversation.ConversationUsageListStatus;
+import ai.z.openapi.service.assistant.query_support.AssistantSupportStatus;
+import ai.z.openapi.service.assistant.conversation.ConversationParameters;
+import ai.z.openapi.service.assistant.query_support.QuerySupportParams;
+import io.reactivex.Single;
+import okhttp3.ResponseBody;
+import retrofit2.Call;
+import retrofit2.http.Body;
+import retrofit2.http.POST;
+import retrofit2.http.Streaming;
+
+public interface AssistantApi {
+
+ @Streaming
+ @POST("assistant")
+ Call assistantCompletionStream(@Body AssistantParameters request);
+
+ @POST("assistant")
+ Single assistantCompletion(@Body AssistantParameters request);
+
+ @POST("assistant/list")
+ Single querySupport(@Body QuerySupportParams request);
+
+ @POST("assistant/conversation/list")
+ Single queryConversationUsage(@Body ConversationParameters request);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/audio/AudioApi.java b/core/src/main/java/ai/z/openapi/api/audio/AudioApi.java
new file mode 100644
index 0000000..f8c05fb
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/audio/AudioApi.java
@@ -0,0 +1,50 @@
+package ai.z.openapi.api.audio;
+
+import ai.z.openapi.service.audio.AudioSpeechRequest;
+import ai.z.openapi.service.model.ModelData;
+import io.reactivex.Single;
+import okhttp3.MultipartBody;
+import okhttp3.RequestBody;
+import okhttp3.ResponseBody;
+
+import retrofit2.Call;
+import retrofit2.http.Body;
+import retrofit2.http.Multipart;
+import retrofit2.http.POST;
+import retrofit2.http.Part;
+import retrofit2.http.PartMap;
+import retrofit2.http.Streaming;
+
+import java.util.Map;
+
+public interface AudioApi {
+
+ /**
+ * TTS interface (Text to speech)
+ * @param request
+ * @return
+ */
+ @POST("audio/speech")
+ Single audioSpeech(@Body AudioSpeechRequest request);
+
+ /**
+ * Voice cloning interface
+ * @param request
+ * @return
+ */
+ @Multipart
+ @POST("audio/customization")
+ Single audioCustomization(@PartMap Map request,
+ @Part MultipartBody.Part voiceData);
+
+ @Streaming
+ @POST("audio/transcriptions")
+ @Multipart
+ Call audioTranscriptionsStream(@PartMap Map request,
+ @Part MultipartBody.Part file);
+
+ @POST("audio/transcriptions")
+ @Multipart
+ Single audioTranscriptions(@PartMap Map request, @Part MultipartBody.Part file);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/batches/BatchesApi.java b/core/src/main/java/ai/z/openapi/api/batches/BatchesApi.java
new file mode 100644
index 0000000..3456d74
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/batches/BatchesApi.java
@@ -0,0 +1,27 @@
+package ai.z.openapi.api.batches;
+
+import ai.z.openapi.service.batches.Batch;
+import ai.z.openapi.service.batches.BatchCreateParams;
+import ai.z.openapi.service.batches.BatchPage;
+import io.reactivex.Single;
+import retrofit2.http.Body;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.Path;
+import retrofit2.http.Query;
+
+public interface BatchesApi {
+
+ @POST("batches")
+ Single batchesCreate(@Body BatchCreateParams batchCreateParams);
+
+ @GET("batches/{batch_id}")
+ Single batchesRetrieve(@Path("batch_id") String batchId);
+
+ @GET("batches")
+ Single batchesList(@Query("after") String after, @Query("limit") Integer limit);
+
+ @POST("batches/{batch_id}/cancel")
+ Single batchesCancel(@Path("batch_id") String batchId);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/chat/ChatApi.java b/core/src/main/java/ai/z/openapi/api/chat/ChatApi.java
new file mode 100644
index 0000000..c89b208
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/chat/ChatApi.java
@@ -0,0 +1,29 @@
+package ai.z.openapi.api.chat;
+
+import ai.z.openapi.service.model.ChatCompletionCreateParams;
+import ai.z.openapi.service.model.ModelData;
+import io.reactivex.Single;
+import okhttp3.ResponseBody;
+import retrofit2.Call;
+import retrofit2.http.Body;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.Path;
+import retrofit2.http.Streaming;
+
+public interface ChatApi {
+
+ @Streaming
+ @POST("chat/completions")
+ Call createChatCompletionStream(@Body ChatCompletionCreateParams request);
+
+ @POST("async/chat/completions")
+ Single createChatCompletionAsync(@Body ChatCompletionCreateParams request);
+
+ @POST("chat/completions")
+ Single createChatCompletion(@Body ChatCompletionCreateParams request);
+
+ @GET("async-result/{id}")
+ Single queryAsyncResult(@Path("id") String id);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/embedding/EmbeddingApi.java b/core/src/main/java/ai/z/openapi/api/embedding/EmbeddingApi.java
new file mode 100644
index 0000000..e41334b
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/embedding/EmbeddingApi.java
@@ -0,0 +1,14 @@
+package ai.z.openapi.api.embedding;
+
+import ai.z.openapi.service.embedding.EmbeddingCreateParams;
+import ai.z.openapi.service.embedding.EmbeddingResult;
+import io.reactivex.Single;
+import retrofit2.http.Body;
+import retrofit2.http.POST;
+
+public interface EmbeddingApi {
+
+ @POST("embeddings")
+ Single createEmbeddings(@Body EmbeddingCreateParams request);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/file/FileApi.java b/core/src/main/java/ai/z/openapi/api/file/FileApi.java
new file mode 100644
index 0000000..a439c24
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/file/FileApi.java
@@ -0,0 +1,37 @@
+package ai.z.openapi.api.file;
+
+import ai.z.openapi.service.file.File;
+import ai.z.openapi.service.file.FileDeleted;
+import ai.z.openapi.service.file.QueryFileResult;
+import io.reactivex.Single;
+import okhttp3.MultipartBody;
+import okhttp3.ResponseBody;
+import retrofit2.Call;
+import retrofit2.http.Body;
+import retrofit2.http.DELETE;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.Path;
+import retrofit2.http.Query;
+import retrofit2.http.Streaming;
+
+public interface FileApi {
+
+ @POST("files")
+ Single uploadFile(@Body MultipartBody multipartBody);
+
+ @GET("files/{file_id}")
+ Single retrieveFile(@Path("file_id") String fileId);
+
+ @DELETE("files/{file_id}")
+ Single deletedFile(@Path("file_id") String fileId);
+
+ @GET("files")
+ Single queryFileList(@Query("after") String after, @Query("purpose") String purpose,
+ @Query("order") String order, @Query("limit") Integer limit);
+
+ @Streaming
+ @GET("files/{file_id}/content")
+ Call fileContent(@Path("file_id") String fileId);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/fine_tuning/FineTuningApi.java b/core/src/main/java/ai/z/openapi/api/fine_tuning/FineTuningApi.java
new file mode 100644
index 0000000..ec3f9a4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/fine_tuning/FineTuningApi.java
@@ -0,0 +1,42 @@
+package ai.z.openapi.api.fine_tuning;
+
+import ai.z.openapi.service.fine_turning.FineTunedModelsStatus;
+import ai.z.openapi.service.fine_turning.FineTuningEvent;
+import ai.z.openapi.service.fine_turning.FineTuningJob;
+import ai.z.openapi.service.fine_turning.FineTuningJobRequest;
+import ai.z.openapi.service.fine_turning.PersonalFineTuningJob;
+import io.reactivex.Single;
+import retrofit2.http.Body;
+import retrofit2.http.DELETE;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.Path;
+import retrofit2.http.Query;
+
+public interface FineTuningApi {
+
+ @POST("fine_tuning/jobs")
+ Single createFineTuningJob(@Body FineTuningJobRequest request);
+
+ @GET("fine_tuning/jobs/{fine_tuning_job_id}/events")
+ Single listFineTuningJobEvents(@Path("fine_tuning_job_id") String fineTuningJobId,
+ @Query("limit") Integer limit, @Query("after") String after);
+
+ @GET("fine_tuning/jobs/{fine_tuning_job_id}")
+ Single retrieveFineTuningJob(@Path("fine_tuning_job_id") String fineTuningJobId,
+ @Query("limit") Integer limit, @Query("after") String after);
+
+ @GET("fine_tuning/jobs")
+ Single queryPersonalFineTuningJobs(@Query("limit") Integer limit,
+ @Query("after") String after);
+
+ @POST("fine_tuning/jobs/{fine_tuning_job_id}/cancel")
+ Single cancelFineTuningJob(@Path("fine_tuning_job_id") String fineTuningJobId);
+
+ @DELETE("fine_tuning/jobs/{fine_tuning_job_id}")
+ Single deleteFineTuningJob(@Path("fine_tuning_job_id") String fineTuningJobId);
+
+ @DELETE("fine_tuning/fine_tuned_models/{fine_tuned_model}")
+ Single deleteFineTuningModel(@Path("fine_tuned_model") String fineTunedModel);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/images/ImagesApi.java b/core/src/main/java/ai/z/openapi/api/images/ImagesApi.java
new file mode 100644
index 0000000..45121be
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/images/ImagesApi.java
@@ -0,0 +1,14 @@
+package ai.z.openapi.api.images;
+
+import ai.z.openapi.service.image.CreateImageRequest;
+import ai.z.openapi.service.image.ImageResult;
+import io.reactivex.Single;
+import retrofit2.http.Body;
+import retrofit2.http.POST;
+
+public interface ImagesApi {
+
+ @POST("images/generations")
+ Single createImage(@Body CreateImageRequest request);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/knowledge/KnowledgeApi.java b/core/src/main/java/ai/z/openapi/api/knowledge/KnowledgeApi.java
new file mode 100644
index 0000000..139df43
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/knowledge/KnowledgeApi.java
@@ -0,0 +1,36 @@
+package ai.z.openapi.api.knowledge;
+
+import ai.z.openapi.service.knowledge.KnowledgeBaseParams;
+import ai.z.openapi.service.knowledge.KnowledgeInfo;
+import ai.z.openapi.service.knowledge.KnowledgePage;
+import ai.z.openapi.service.knowledge.KnowledgeUsed;
+import io.reactivex.Single;
+
+import retrofit2.Response;
+import retrofit2.http.Body;
+import retrofit2.http.DELETE;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.PUT;
+import retrofit2.http.Path;
+import retrofit2.http.Query;
+
+public interface KnowledgeApi {
+
+ @POST("knowledge")
+ Single knowledgeCreate(@Body KnowledgeBaseParams knowledgeBaseParams);
+
+ @PUT("knowledge/{knowledge_id}")
+ Single> knowledgeModify(@Path("knowledge_id") String knowledge_id,
+ @Body KnowledgeBaseParams knowledgeBaseParams);
+
+ @GET("knowledge")
+ Single knowledgeQuery(@Query("page") Integer page, @Query("size") Integer size);
+
+ @DELETE("knowledge/{knowledge_id}")
+ Single> knowledgeDelete(@Path("knowledge_id") String knowledge_id);
+
+ @GET("knowledge/capacity")
+ Single knowledgeUsed();
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/knowledge/document/DocumentApi.java b/core/src/main/java/ai/z/openapi/api/knowledge/document/DocumentApi.java
new file mode 100644
index 0000000..f50b029
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/knowledge/document/DocumentApi.java
@@ -0,0 +1,37 @@
+package ai.z.openapi.api.knowledge.document;
+
+import ai.z.openapi.service.knowledge.document.DocumentData;
+import ai.z.openapi.service.knowledge.document.DocumentEditParams;
+import ai.z.openapi.service.knowledge.document.DocumentObject;
+import ai.z.openapi.service.knowledge.document.DocumentPage;
+import io.reactivex.Single;
+import okhttp3.MultipartBody;
+import retrofit2.Response;
+import retrofit2.http.Body;
+import retrofit2.http.DELETE;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.PUT;
+import retrofit2.http.Path;
+import retrofit2.http.Query;
+
+public interface DocumentApi {
+
+ @POST("files")
+ Single createDocument(@Body MultipartBody document);
+
+ @PUT("document/{document_id}")
+ Single> modifyDocument(@Path("document_id") String documentId,
+ @Body DocumentEditParams documentEditParams);
+
+ @DELETE("document/{document_id}")
+ Single> deleteDocument(@Path("document_id") String documentId);
+
+ @GET("files")
+ Single queryDocumentList(@Query("knowledge_id") String knowledgeId, @Query("purpose") String purpose,
+ @Query("page") Integer page, @Query("limit") Integer limit, @Query("order") String order);
+
+ @GET("document/{document_id}")
+ Single retrieveDocument(@Path("document_id") String documentId);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/tools/ToolsApi.java b/core/src/main/java/ai/z/openapi/api/tools/ToolsApi.java
new file mode 100644
index 0000000..a2b9683
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/tools/ToolsApi.java
@@ -0,0 +1,21 @@
+package ai.z.openapi.api.tools;
+
+import ai.z.openapi.service.tools.WebSearchParamsRequest;
+import ai.z.openapi.service.tools.WebSearchPro;
+import io.reactivex.Single;
+import okhttp3.ResponseBody;
+import retrofit2.Call;
+import retrofit2.http.Body;
+import retrofit2.http.POST;
+import retrofit2.http.Streaming;
+
+public interface ToolsApi {
+
+ @Streaming
+ @POST("tools")
+ Call webSearchStreaming(@Body WebSearchParamsRequest request);
+
+ @POST("tools")
+ Single webSearch(@Body WebSearchParamsRequest request);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/videos/VideosApi.java b/core/src/main/java/ai/z/openapi/api/videos/VideosApi.java
new file mode 100644
index 0000000..3327a2f
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/videos/VideosApi.java
@@ -0,0 +1,19 @@
+package ai.z.openapi.api.videos;
+
+import ai.z.openapi.service.videos.VideoCreateParams;
+import ai.z.openapi.service.videos.VideoObject;
+import io.reactivex.Single;
+import retrofit2.http.Body;
+import retrofit2.http.GET;
+import retrofit2.http.POST;
+import retrofit2.http.Path;
+
+public interface VideosApi {
+
+ @POST("videos/generations")
+ Single videoGenerations(@Body VideoCreateParams request);
+
+ @GET("async-result/{id}")
+ Single videoGenerationsResult(@Path("id") String id);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/api/web_search/WebSearchApi.java b/core/src/main/java/ai/z/openapi/api/web_search/WebSearchApi.java
new file mode 100644
index 0000000..f1728ea
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/api/web_search/WebSearchApi.java
@@ -0,0 +1,14 @@
+package ai.z.openapi.api.web_search;
+
+import ai.z.openapi.service.web_search.WebSearchDTO;
+import ai.z.openapi.service.web_search.WebSearchRequest;
+import io.reactivex.Single;
+import retrofit2.http.Body;
+import retrofit2.http.POST;
+
+public interface WebSearchApi {
+
+ @POST("web_search")
+ Single webSearch(@Body WebSearchRequest request);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/Constants.java b/core/src/main/java/ai/z/openapi/core/Constants.java
new file mode 100644
index 0000000..e9a29f4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/Constants.java
@@ -0,0 +1,155 @@
+package ai.z.openapi.core;
+
+/**
+ * Constants class containing all the configuration values and model identifiers used
+ * throughout the Z.AI OpenAPI SDK.
+ *
+ * This class provides centralized access to: - API base URLs - Model identifiers for
+ * different AI capabilities - Invocation method constants
+ *
+ * @author Z.AI SDK Team
+ * @since 1.0.0
+ */
+public final class Constants {
+
+ // Private constructor to prevent instantiation
+ private Constants() {
+ throw new UnsupportedOperationException("Constants class cannot be instantiated");
+ }
+
+ // =============================================================================
+ // API Configuration
+ // =============================================================================
+
+ /**
+ * Base URL for the ZHIPU AI OpenAPI service. All API requests will be made to
+ * endpoints under this base URL.
+ */
+ public static final String ZHIPU_AI_BASE_URL = "https://open.bigmodel.cn/api/paas/v4/";
+
+ /**
+ * Base URL for the Z.AI OpenAPI service. All API requests will be made to endpoints
+ * under this base URL.
+ */
+ public static final String Z_AI_BASE_URL = "https://api.z.ai/api/paas/v4/";
+
+ // =============================================================================
+ // Text Generation Models
+ // =============================================================================
+
+ /**
+ * GLM-4 Plus model - Enhanced version with improved capabilities.
+ */
+ public static final String ModelChatGLM4Plus = "glm-4-plus";
+
+ /**
+ * GLM-4 Air model - Lightweight version optimized for speed.
+ */
+ public static final String ModelChatGLM4Air = "glm-4-air";
+
+ /**
+ * GLM-4 Flash model - Ultra-fast response model.
+ */
+ public static final String ModelChatGLM4Flash = "glm-4-flash";
+
+ /**
+ * GLM-4 standard model - Balanced performance and capability.
+ */
+ public static final String ModelChatGLM4 = "glm-4";
+
+ /**
+ * GLM-4 model version 0520 - Specific version release.
+ */
+ public static final String ModelChatGLM40520 = "glm-4-0520";
+
+ /**
+ * GLM-4 AirX model - Extended Air model with additional features.
+ */
+ public static final String ModelChatGLM4Airx = "glm-4-airx";
+
+ /**
+ * GLM-4 Long model - Optimized for long-context conversations.
+ */
+ public static final String ModelChatGLMLong = "glm-4-long";
+
+ /**
+ * GLM-4 Voice model - Specialized for voice-related tasks.
+ */
+ public static final String ModelChatGLM4Voice = "glm-4-voice";
+
+ // =============================================================================
+ // Vision Models (Image Understanding)
+ // =============================================================================
+
+ /**
+ * GLM-4V Plus model - Enhanced vision model for image understanding.
+ */
+ public static final String ModelChatGLM4VPlus = "glm-4v-plus";
+
+ /**
+ * GLM-4V standard model - Standard vision model for image analysis.
+ */
+ public static final String ModelChatGLM4V = "glm-4v";
+
+ // =============================================================================
+ // Image Generation Models
+ // =============================================================================
+
+ /**
+ * CogView-3 Plus model - Enhanced image generation capabilities.
+ */
+ public static final String ModelCogView3Plus = "cogview-3-plus";
+
+ /**
+ * CogView-3 standard model - Standard image generation model.
+ */
+ public static final String ModelCogView = "cogview-3";
+
+ // =============================================================================
+ // Embedding Models
+ // =============================================================================
+
+ /**
+ * Embedding model version 2 - Text embedding generation.
+ */
+ public static final String ModelEmbedding2 = "embedding-2";
+
+ /**
+ * Embedding model version 3 - Latest text embedding generation.
+ */
+ public static final String ModelEmbedding3 = "embedding-3";
+
+ // =============================================================================
+ // Specialized Models
+ // =============================================================================
+
+ /**
+ * CharGLM-3 model - Anthropomorphic character interaction model.
+ */
+ public static final String ModelCharGLM3 = "charglm-3";
+
+ /**
+ * CogTTS model - Text-to-Speech synthesis model.
+ */
+ public static final String ModelTTS = "cogtts";
+
+ // =============================================================================
+ // API Invocation Methods
+ // =============================================================================
+
+ /**
+ * Asynchronous invocation method - For non-blocking API calls.
+ */
+ public static final String INVOKE_METHOD_ASYNC = "async-invoke";
+
+ /**
+ * Server-Sent Events invocation method - For streaming responses.
+ */
+ public static final String INVOKE_METHOD_SSE = "sse-invoke";
+
+ /**
+ * Standard synchronous invocation method - For blocking API calls.
+ */
+ public static final String INVOKE_METHOD = "invoke";
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/cache/ICache.java b/core/src/main/java/ai/z/openapi/core/cache/ICache.java
new file mode 100644
index 0000000..c130839
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/cache/ICache.java
@@ -0,0 +1,27 @@
+package ai.z.openapi.core.cache;
+
+import java.util.concurrent.TimeUnit;
+
+/**
+ * Token cache interface with default LocalCache implementation. Can be replaced with
+ * distributed cache (e.g., Redis) as needed.
+ */
+public interface ICache {
+
+ /**
+ * Retrieves cached value by key.
+ * @param key the cache key
+ * @return cached value or empty string if not found or expired
+ */
+ String get(String key);
+
+ /**
+ * Sets cache value with expiration time.
+ * @param key the cache key
+ * @param value the value to cache
+ * @param expire expiration duration
+ * @param timeUnit time unit for expiration
+ */
+ void set(String key, String value, int expire, TimeUnit timeUnit);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/cache/LocalCache.java b/core/src/main/java/ai/z/openapi/core/cache/LocalCache.java
new file mode 100644
index 0000000..ea73457
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/cache/LocalCache.java
@@ -0,0 +1,81 @@
+package ai.z.openapi.core.cache;
+
+import java.util.Calendar;
+import java.util.Date;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+import java.util.concurrent.TimeUnit;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Thread-safe local cache implementation using ConcurrentHashMap. Provides basic caching
+ * functionality with expiration support.
+ */
+public class LocalCache implements ICache {
+
+ private static final Logger log = LoggerFactory.getLogger(LocalCache.class);
+
+ private static final ConcurrentMap CACHE = new ConcurrentHashMap<>(8);
+
+ /**
+ * Private constructor to prevent direct instantiation.
+ */
+ private LocalCache() {
+ }
+
+ /**
+ * Gets singleton instance of LocalCache.
+ * @return LocalCache instance
+ */
+ public static LocalCache getInstance() {
+ return Inner.LOCAL_CACHE;
+ }
+
+ @Override
+ public String get(String key) {
+ Value v = LocalCache.CACHE.get(key);
+ if (v == null || new Date().after(v.end)) {
+ return "";
+ }
+
+ log.debug("Retrieved key: {}, time left: {}s", key, (v.end.getTime() - new Date().getTime()) / 1000);
+ return v.value;
+ }
+
+ @Override
+ public void set(String key, String value, int expire, TimeUnit timeUnit) {
+ Calendar calendar = Calendar.getInstance();
+ calendar.add(Calendar.SECOND, (int) timeUnit.toSeconds(expire));
+ Value v = new Value(value, calendar.getTime());
+ log.debug("Cached key: {}, expire time: {}", key, calendar.getTime());
+ LocalCache.CACHE.put(key, v);
+ }
+
+ /**
+ * Internal value wrapper with expiration time.
+ */
+ private static class Value {
+
+ final String value;
+
+ final Date end;
+
+ public Value(String value, Date time) {
+ this.value = value;
+ this.end = time;
+ }
+
+ }
+
+ /**
+ * Singleton holder pattern for thread-safe lazy initialization.
+ */
+ private static class Inner {
+
+ private static final LocalCache LOCAL_CACHE = new LocalCache();
+
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/config/ZaiConfig.java b/core/src/main/java/ai/z/openapi/core/config/ZaiConfig.java
new file mode 100644
index 0000000..8405bf9
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/config/ZaiConfig.java
@@ -0,0 +1,400 @@
+package ai.z.openapi.core.config;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+import java.util.concurrent.TimeUnit;
+
+import static ai.z.openapi.core.Constants.Z_AI_BASE_URL;
+
+/**
+ * Configuration class for ZAI SDK containing API credentials, JWT settings, HTTP client
+ * configurations, and cache settings. Supports reading configuration values from
+ * environment variables with memory values taking priority.
+ */
+@Setter
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class ZaiConfig {
+
+ // Environment variable names
+ private static final String ENV_BASE_URL = "ZAI_BASE_URL";
+
+ private static final String ENV_API_KEY = "ZAI_API_KEY";
+
+ private static final String ENV_EXPIRE_MILLIS = "ZAI_EXPIRE_MILLIS";
+
+ private static final String ENV_ALG = "ZAI_ALG";
+
+ private static final String ENV_DISABLE_TOKEN_CACHE = "ZAI_DISABLE_TOKEN_CACHE";
+
+ private static final String ENV_CONNECTION_POOL_MAX_IDLE = "ZAI_CONNECTION_POOL_MAX_IDLE";
+
+ private static final String ENV_CONNECTION_POOL_KEEP_ALIVE = "ZAI_CONNECTION_POOL_KEEP_ALIVE";
+
+ private static final String ENV_REQUEST_TIMEOUT = "ZAI_REQUEST_TIMEOUT";
+
+ private static final String ENV_CONNECT_TIMEOUT = "ZAI_CONNECT_TIMEOUT";
+
+ private static final String ENV_READ_TIMEOUT = "ZAI_READ_TIMEOUT";
+
+ private static final String ENV_WRITE_TIMEOUT = "ZAI_WRITE_TIMEOUT";
+
+ /**
+ * Base URL for API endpoints.
+ */
+ private String baseUrl;
+
+ /**
+ * Combined API secret key in format: {apiId}.{apiSecret}
+ */
+ private String apiKey;
+
+ /**
+ * API id component.
+ */
+ private String apiId;
+
+ /**
+ * API secret component.
+ */
+ private String apiSecret;
+
+ /**
+ * JWT token expiration time in milliseconds (default: 30 minutes).
+ */
+ private int expireMillis = 30 * 60 * 1000;
+
+ /**
+ * JWT encryption algorithm (default: HS256).
+ */
+ private String alg = "HS256";
+
+ /**
+ * Flag to disable token caching.
+ */
+ private boolean disableTokenCache;
+
+ /**
+ * Maximum number of idle connections in the connection pool.
+ */
+ private int connectionPoolMaxIdleConnections = 5;
+
+ /**
+ * Keep alive duration for connections in the pool (in seconds).
+ */
+ private long connectionPoolKeepAliveDuration = 1;
+
+ /**
+ * Time unit for connection pool keep alive duration.
+ */
+ private TimeUnit connectionPoolTimeUnit = TimeUnit.SECONDS;
+
+ /**
+ * Request timeout in specified time unit.
+ */
+ private int requestTimeOut = 300;
+
+ /**
+ * Connection timeout in specified time unit.
+ */
+ private int connectTimeout = 100;
+
+ /**
+ * Read timeout in specified time unit.
+ */
+ private int readTimeout = 100;
+
+ /**
+ * Write timeout in specified time unit.
+ */
+ private int writeTimeout = 100;
+
+ /**
+ * Time unit for timeout configurations.
+ */
+ private TimeUnit timeOutTimeUnit = TimeUnit.SECONDS;
+
+ /**
+ * Source channel identifier for request tracking.
+ */
+ private String source_channel = "java-sdk";
+
+ /**
+ * Constructor with combined API secret key.
+ * @param apiKey combined secret key in format {apiKey}.{apiSecret}
+ * @throws RuntimeException if apiSecretKey format is invalid
+ */
+ public ZaiConfig(String apiKey) {
+ this.apiKey = apiKey;
+ String[] arrStr = apiKey.split("\\.");
+ if (arrStr.length != 2) {
+ throw new RuntimeException("invalid apiSecretKey");
+ }
+ this.apiId = arrStr[0];
+ this.apiSecret = arrStr[1];
+ }
+
+ /**
+ * Sets API key and parses it into id and secret components.
+ * @param apiKey combined secret key in format {apiId}.{apiSecret}
+ * @throws RuntimeException if apiSecretKey format is invalid
+ */
+ public void setApiKey(String apiKey) {
+ this.apiKey = apiKey;
+ String[] arrStr = apiKey.split("\\.");
+ if (arrStr.length != 2) {
+ throw new RuntimeException("invalid api Key");
+ }
+ this.apiId = arrStr[0];
+ this.apiSecret = arrStr[1];
+ }
+
+ /**
+ * Gets base URL with system property and environment variable fallback.
+ */
+ public String getBaseUrl() {
+ if (baseUrl != null) {
+ return baseUrl;
+ }
+ String propValue = System.getProperty(ENV_BASE_URL);
+ propValue = propValue != null ? propValue : System.getenv(ENV_BASE_URL);
+ return propValue != null ? propValue : Z_AI_BASE_URL;
+ }
+
+ /**
+ * Gets API secret key with system property and environment variable fallback.
+ */
+ public String getApiKey() {
+ if (apiKey != null && !apiKey.isEmpty()) {
+ return apiKey;
+ }
+ String propValue = System.getProperty(ENV_API_KEY);
+ String value = propValue != null ? propValue : System.getenv(ENV_API_KEY);
+ if (value != null && !value.isEmpty()) {
+ // Parse value and set components
+ this.apiKey = value;
+ String[] arrStr = value.split("\\.");
+ if (arrStr.length == 2) {
+ this.apiId = arrStr[0];
+ this.apiSecret = arrStr[1];
+ }
+ return value;
+ }
+ return apiKey;
+ }
+
+ /**
+ * Gets API key with system property and environment variable fallback.
+ */
+ public String getApiId() {
+ if (apiId != null && !apiId.isEmpty()) {
+ return apiId;
+ }
+ getApiKey();
+ return apiId;
+ }
+
+ /**
+ * Gets API secret with system property and environment variable fallback.
+ */
+ public String getApiSecret() {
+ if (apiSecret != null && !apiSecret.isEmpty()) {
+ return apiSecret;
+ }
+ getApiKey();
+ return apiSecret;
+ }
+
+ /**
+ * Gets expire millis with system property and environment variable fallback.
+ */
+ public int getExpireMillis() {
+ if (expireMillis != 30 * 60 * 1000) { // If not default value
+ return expireMillis;
+ }
+ String propValue = System.getProperty(ENV_EXPIRE_MILLIS);
+ String value = propValue != null ? propValue : System.getenv(ENV_EXPIRE_MILLIS);
+ if (value != null) {
+ try {
+ return Integer.parseInt(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return expireMillis;
+ }
+
+ /**
+ * Gets algorithm with system property and environment variable fallback.
+ */
+ public String getAlg() {
+ if (!"HS256".equals(alg)) {
+ return alg;
+ }
+ String propValue = System.getProperty(ENV_ALG);
+ String value = propValue != null ? propValue : System.getenv(ENV_ALG);
+ return value != null ? value : alg;
+ }
+
+ /**
+ * Gets disable token cache flag with system property and environment variable
+ * fallback.
+ */
+ public boolean isDisableTokenCache() {
+ if (disableTokenCache) {
+ return true;
+ }
+ String propValue = System.getProperty(ENV_DISABLE_TOKEN_CACHE);
+ String value = propValue != null ? propValue : System.getenv(ENV_DISABLE_TOKEN_CACHE);
+ return Boolean.parseBoolean(value);
+ }
+
+ /**
+ * Gets connection pool max idle connections with system property and environment
+ * variable fallback.
+ */
+ public int getConnectionPoolMaxIdleConnections() {
+ if (connectionPoolMaxIdleConnections != 5) { // If not default value
+ return connectionPoolMaxIdleConnections;
+ }
+ String propValue = System.getProperty(ENV_CONNECTION_POOL_MAX_IDLE);
+ String value = propValue != null ? propValue : System.getenv(ENV_CONNECTION_POOL_MAX_IDLE);
+ if (value != null) {
+ try {
+ return Integer.parseInt(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return connectionPoolMaxIdleConnections;
+ }
+
+ /**
+ * Gets connection pool keep alive duration with system property and environment
+ * variable fallback.
+ */
+ public long getConnectionPoolKeepAliveDuration() {
+ if (connectionPoolKeepAliveDuration != 1) { // If not default value
+ return connectionPoolKeepAliveDuration;
+ }
+ String propValue = System.getProperty(ENV_CONNECTION_POOL_KEEP_ALIVE);
+ String value = propValue != null ? propValue : System.getenv(ENV_CONNECTION_POOL_KEEP_ALIVE);
+ if (value != null) {
+ try {
+ return Long.parseLong(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return connectionPoolKeepAliveDuration;
+ }
+
+ /**
+ * Gets connection pool time unit (always returns the set value or default).
+ */
+ public TimeUnit getConnectionPoolTimeUnit() {
+ return connectionPoolTimeUnit != null ? connectionPoolTimeUnit : TimeUnit.SECONDS;
+ }
+
+ /**
+ * Gets request timeout with system property and environment variable fallback.
+ */
+ public int getRequestTimeOut() {
+ if (requestTimeOut != 300) {
+ return requestTimeOut;
+ }
+ String propValue = System.getProperty(ENV_REQUEST_TIMEOUT);
+ String value = propValue != null ? propValue : System.getenv(ENV_REQUEST_TIMEOUT);
+ if (value != null) {
+ try {
+ return Integer.parseInt(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return requestTimeOut;
+ }
+
+ /**
+ * Gets connect timeout with system property and environment variable fallback.
+ */
+ public int getConnectTimeout() {
+ if (connectTimeout != 100) {
+ return connectTimeout;
+ }
+ String propValue = System.getProperty(ENV_CONNECT_TIMEOUT);
+ String value = propValue != null ? propValue : System.getenv(ENV_CONNECT_TIMEOUT);
+ if (value != null) {
+ try {
+ return Integer.parseInt(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return connectTimeout;
+ }
+
+ /**
+ * Gets read timeout with system property and environment variable fallback.
+ */
+ public int getReadTimeout() {
+ if (readTimeout != 100) {
+ return readTimeout;
+ }
+ String propValue = System.getProperty(ENV_READ_TIMEOUT);
+ String value = propValue != null ? propValue : System.getenv(ENV_READ_TIMEOUT);
+ if (value != null) {
+ try {
+ return Integer.parseInt(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return readTimeout;
+ }
+
+ /**
+ * Gets write timeout with system property and environment variable fallback.
+ */
+ public int getWriteTimeout() {
+ if (writeTimeout != 100) {
+ return writeTimeout;
+ }
+ String propValue = System.getProperty(ENV_WRITE_TIMEOUT);
+ String value = propValue != null ? propValue : System.getenv(ENV_WRITE_TIMEOUT);
+ if (value != null) {
+ try {
+ return Integer.parseInt(value);
+ }
+ catch (NumberFormatException e) {
+ // Return default value if parsing fails
+ }
+ }
+ return writeTimeout;
+ }
+
+ /**
+ * Gets timeout time unit (always returns the set value or null).
+ */
+ public TimeUnit getTimeOutTimeUnit() {
+ return timeOutTimeUnit;
+ }
+
+ /**
+ * Gets source channel with system property.
+ */
+ public String getSource_channel() {
+ return source_channel;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/model/ClientRequest.java b/core/src/main/java/ai/z/openapi/core/model/ClientRequest.java
new file mode 100644
index 0000000..601f8d0
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/model/ClientRequest.java
@@ -0,0 +1,5 @@
+package ai.z.openapi.core.model;
+
+public interface ClientRequest {
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/model/ClientResponse.java b/core/src/main/java/ai/z/openapi/core/model/ClientResponse.java
new file mode 100644
index 0000000..8b0ef37
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/model/ClientResponse.java
@@ -0,0 +1,19 @@
+package ai.z.openapi.core.model;
+
+import ai.z.openapi.service.model.ChatError;
+
+public interface ClientResponse {
+
+ T getData();
+
+ void setData(T data);
+
+ void setCode(int code);
+
+ void setMsg(String msg);
+
+ void setSuccess(boolean b);
+
+ void setError(ChatError chatError);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/model/FlowableClientResponse.java b/core/src/main/java/ai/z/openapi/core/model/FlowableClientResponse.java
new file mode 100644
index 0000000..d995477
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/model/FlowableClientResponse.java
@@ -0,0 +1,19 @@
+package ai.z.openapi.core.model;
+
+import io.reactivex.Flowable;
+
+/**
+ * Client response interface with reactive stream support. Extends ClientResponse to
+ * provide Flowable stream functionality.
+ *
+ * @param response data type
+ */
+public interface FlowableClientResponse extends ClientResponse {
+
+ /**
+ * Sets the reactive stream for this response.
+ * @param stream Flowable stream containing response data
+ */
+ void setFlowable(Flowable stream);
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/response/HttpxBinaryResponseContent.java b/core/src/main/java/ai/z/openapi/core/response/HttpxBinaryResponseContent.java
new file mode 100644
index 0000000..e2e3537
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/response/HttpxBinaryResponseContent.java
@@ -0,0 +1,156 @@
+package ai.z.openapi.core.response;
+
+import java.io.Closeable;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.Iterator;
+import java.util.Objects;
+import okhttp3.ResponseBody;
+import okio.BufferedSource;
+
+public class HttpxBinaryResponseContent implements Closeable {
+
+ private final retrofit2.Response response;
+
+ public HttpxBinaryResponseContent(retrofit2.Response response) {
+ if (response == null || response.body() == null) {
+ throw new IllegalArgumentException("Response or ResponseBody cannot be null");
+ }
+ this.response = response;
+ }
+
+ public byte[] getContent() throws IOException {
+ if (response.body() == null) {
+ throw new IOException("ResponseBody is null");
+ }
+ try (BufferedSource source = response.body().source()) {
+ return source.readByteArray();
+ }
+ }
+
+ public String getText() throws IOException {
+ if (response.body() == null) {
+ throw new IOException("ResponseBody is null");
+ }
+ try (BufferedSource source = response.body().source()) {
+ return source.readUtf8();
+ }
+ }
+
+ public String getEncoding() {
+ return response.body() != null && response.body().contentType() != null
+ ? Objects.requireNonNull(Objects.requireNonNull(response.body().contentType()).charset()).toString()
+ : null;
+ }
+
+ public Iterator iterBytes(int chunkSize) throws IOException {
+ if (response.body() == null) {
+ throw new IOException("ResponseBody is null");
+ }
+ return new Iterator() {
+ final BufferedSource source = response.body().source();
+
+ final byte[] buffer = new byte[chunkSize];
+
+ boolean hasMore = true;
+
+ @Override
+ public boolean hasNext() {
+ try {
+ if (source.exhausted()) {
+ hasMore = false;
+ }
+ }
+ catch (IOException e) {
+ hasMore = false;
+ }
+ return hasMore;
+ }
+
+ @Override
+ public byte[] next() {
+ try {
+ source.read(buffer);
+ }
+ catch (IOException e) {
+ // Handle the exception
+ }
+ return buffer;
+ }
+ };
+ }
+
+ public Iterator iterText(int chunkSize) throws IOException {
+ if (response.body() == null) {
+ throw new IOException("ResponseBody is null");
+ }
+ return new Iterator() {
+ final BufferedSource source = response.body().source();
+
+ final byte[] buffer = new byte[chunkSize];
+
+ boolean hasMore = true;
+
+ @Override
+ public boolean hasNext() {
+ try {
+ if (source.exhausted()) {
+ hasMore = false;
+ }
+ }
+ catch (IOException e) {
+ hasMore = false;
+ }
+ return hasMore;
+ }
+
+ @Override
+ public String next() {
+ try {
+ int bytesRead = source.read(buffer);
+ return new String(buffer, 0, bytesRead);
+ }
+ catch (IOException e) {
+ // Handle the exception
+ }
+ return "";
+ }
+ };
+ }
+
+ public void writeToFile(String file) throws IOException {
+ if (response.body() == null) {
+ throw new IOException("ResponseBody is null");
+ }
+
+ try (BufferedSource source = response.body().source(); FileOutputStream fos = new FileOutputStream(file)) {
+ byte[] buffer = new byte[4096];
+ int bytesRead;
+ while ((bytesRead = source.read(buffer)) != -1) {
+ fos.write(buffer, 0, bytesRead);
+ }
+ }
+ }
+
+ public void streamToFile(String file, int chunkSize) throws IOException {
+ if (response.body() == null) {
+ throw new IOException("ResponseBody is null");
+ }
+
+ try (BufferedSource source = response.body().source(); FileOutputStream fos = new FileOutputStream(file)) {
+ byte[] buffer = new byte[chunkSize];
+ int bytesRead;
+ while ((bytesRead = source.read(buffer)) != -1) {
+ fos.write(buffer, 0, bytesRead);
+ }
+ }
+ }
+
+ @Override
+ public void close() throws IOException {
+ if (response.body() != null) {
+ response.body().close();
+ }
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/token/AuthenticationInterceptor.java b/core/src/main/java/ai/z/openapi/core/token/AuthenticationInterceptor.java
new file mode 100644
index 0000000..322da34
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/token/AuthenticationInterceptor.java
@@ -0,0 +1,47 @@
+package ai.z.openapi.core.token;
+
+import ai.z.openapi.core.config.ZaiConfig;
+import ai.z.openapi.utils.StringUtils;
+import okhttp3.Interceptor;
+import okhttp3.Request;
+import okhttp3.Response;
+
+import java.io.IOException;
+import java.util.Objects;
+
+/**
+ * OkHttp Interceptor that adds an authorization token header
+ */
+public class AuthenticationInterceptor implements Interceptor {
+
+ private final ZaiConfig config;
+
+ public AuthenticationInterceptor(ZaiConfig config) {
+ Objects.requireNonNull(config.getApiKey(), "Z.ai token required");
+ this.config = config;
+ }
+
+ @Override
+ public Response intercept(Chain chain) throws IOException {
+ String accessToken;
+ if (this.config.isDisableTokenCache()) {
+ accessToken = this.config.getApiKey();
+ }
+ else {
+ TokenManager tokenManager = GlobalTokenManager.getTokenManagerV4();
+ accessToken = tokenManager.getToken(this.config);
+ }
+ String source_channel = "java-sdk";
+ if (StringUtils.isNotEmpty(config.getSource_channel())) {
+ source_channel = config.getSource_channel();
+ }
+ Request request = chain.request()
+ .newBuilder()
+ .header("Authorization", "Bearer " + accessToken)
+ .header("x-source-channel", source_channel)
+ .header("Accept-Language", "en-US,en")
+ .build();
+ return chain.proceed(request);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/token/GlobalTokenManager.java b/core/src/main/java/ai/z/openapi/core/token/GlobalTokenManager.java
new file mode 100644
index 0000000..05a6b63
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/token/GlobalTokenManager.java
@@ -0,0 +1,32 @@
+package ai.z.openapi.core.token;
+
+import ai.z.openapi.core.cache.LocalCache;
+
+/**
+ * Global token manager providing singleton access to TokenManager instance. Uses
+ * LocalCache as default cache implementation.
+ */
+public class GlobalTokenManager {
+
+ /**
+ * Global TokenManager instance with LocalCache.
+ */
+ private static volatile TokenManager globalTokenManager = new TokenManager(LocalCache.getInstance());
+
+ /**
+ * Gets the global TokenManager instance.
+ * @return TokenManager instance
+ */
+ public static TokenManager getTokenManagerV4() {
+ return globalTokenManager;
+ }
+
+ /**
+ * Sets custom TokenManager implementation.
+ * @param tokenManager custom TokenManager instance
+ */
+ public static void setTokenManager(TokenManager tokenManager) {
+ globalTokenManager = tokenManager;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/core/token/TokenManager.java b/core/src/main/java/ai/z/openapi/core/token/TokenManager.java
new file mode 100644
index 0000000..33a24f9
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/core/token/TokenManager.java
@@ -0,0 +1,100 @@
+package ai.z.openapi.core.token;
+
+import com.auth0.jwt.JWT;
+import com.auth0.jwt.algorithms.Algorithm;
+import ai.z.openapi.core.config.ZaiConfig;
+import ai.z.openapi.core.cache.ICache;
+import ai.z.openapi.utils.StringUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.nio.charset.StandardCharsets;
+import java.util.Calendar;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * JWT token manager for handling token generation and caching.
+ */
+public class TokenManager {
+
+ private static final Logger logger = LoggerFactory.getLogger(TokenManager.class);
+
+ private final ICache cache;
+
+ private static final String TOKEN_KEY_PREFIX = "zai_oapi_token";
+
+ /**
+ * Additional delay time (5 minutes) to prevent token expiration issues.
+ */
+ private static final Long DELAY_EXPIRE_TIME = 5 * 60 * 1000L;
+
+ /**
+ * Constructs TokenManager with specified cache implementation.
+ * @param cache cache implementation for token storage
+ */
+ public TokenManager(ICache cache) {
+ this.cache = cache;
+ }
+
+ /**
+ * Gets valid JWT token, either from cache or generates new one.
+ * @param config ZAI configuration containing API credentials
+ * @return valid JWT token
+ */
+ public String getToken(ZaiConfig config) {
+ String tokenCacheKey = genTokenCacheKey(config.getApiId());
+ String cacheToken = cache.get(tokenCacheKey);
+ if (StringUtils.isNotEmpty(cacheToken)) {
+ return cacheToken;
+ }
+ String newToken = createJwt(config);
+ cache.set(tokenCacheKey, newToken, config.getExpireMillis(), TimeUnit.MILLISECONDS);
+ return newToken;
+ }
+
+ /**
+ * Creates JWT token using HMAC256 algorithm.
+ * @param config ZAI configuration
+ * @return JWT token string or null if creation fails
+ */
+ private static String createJwt(ZaiConfig config) {
+ Algorithm alg;
+ String algId = config.getAlg();
+ if ("HS256".equals(algId)) {
+ try {
+ alg = Algorithm.HMAC256(config.getApiSecret().getBytes(StandardCharsets.UTF_8));
+ }
+ catch (Exception e) {
+ logger.error("Failed to create HMAC256 algorithm", e);
+ return null;
+ }
+ }
+ else {
+ // Currently only HS256 is supported
+ logger.error("Algorithm: {} not supported", algId);
+ return null;
+ }
+
+ Map payload = new HashMap<>();
+ // here the api_key is the apiId
+ payload.put("api_key", config.getApiId());
+ payload.put("exp", System.currentTimeMillis() + config.getExpireMillis() + DELAY_EXPIRE_TIME);
+ payload.put("timestamp", Calendar.getInstance().getTimeInMillis());
+ Map headerClaims = new HashMap<>();
+ headerClaims.put("alg", "HS256");
+ headerClaims.put("sign_type", "SIGN");
+ return JWT.create().withPayload(payload).withHeader(headerClaims).sign(alg);
+ }
+
+ /**
+ * Generates cache key for token storage.
+ * @param apiKey API key
+ * @return formatted cache key
+ */
+ private String genTokenCacheKey(String apiKey) {
+ return String.format("%s-%s", TOKEN_KEY_PREFIX, apiKey);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/AbstractClientBaseService.java b/core/src/main/java/ai/z/openapi/service/AbstractClientBaseService.java
new file mode 100644
index 0000000..27c6864
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/AbstractClientBaseService.java
@@ -0,0 +1,147 @@
+package ai.z.openapi.service;
+
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.core.model.FlowableClientResponse;
+import ai.z.openapi.service.deserialize.MessageDeserializeFactory;
+import ai.z.openapi.service.model.ResponseBodyCallback;
+import ai.z.openapi.service.model.SSE;
+import ai.z.openapi.service.model.ZAiError;
+import ai.z.openapi.service.model.ZAiHttpException;
+import ai.z.openapi.utils.FlowableRequestSupplier;
+import ai.z.openapi.utils.RequestSupplier;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import io.reactivex.BackpressureStrategy;
+import io.reactivex.Flowable;
+import io.reactivex.Single;
+import okhttp3.ResponseBody;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import retrofit2.Call;
+import retrofit2.HttpException;
+import retrofit2.Response;
+
+import java.io.IOException;
+
+/**
+ * Abstract base service class that provides common functionality for API client
+ * implementations. This class handles request execution, response processing, and error
+ * handling for both synchronous and streaming API calls.
+ */
+public abstract class AbstractClientBaseService {
+
+ protected final static Logger logger = LoggerFactory.getLogger(AbstractClientBaseService.class);
+
+ protected static final ObjectMapper mapper = MessageDeserializeFactory.defaultObjectMapper();
+
+ /**
+ * Executes a synchronous API request.
+ * @param the type of data returned by the API
+ * @param the type of parameters sent to the API
+ * @param the type of the request object
+ * @param the type of the response object
+ * @param request the request object containing parameters
+ * @param requestSupplier the supplier that creates the API call
+ * @param tRespClass the class of the response type
+ * @return the response object containing the API result
+ */
+ public abstract , TResp extends ClientResponse> TResp executeRequest(
+ TReq request, RequestSupplier requestSupplier, Class tRespClass);
+
+ /**
+ * Executes a streaming API request that returns a continuous stream of data.
+ * @param the type of data returned by the API stream
+ * @param the type of parameters sent to the API
+ * @param the type of the request object
+ * @param the type of the streaming response object
+ * @param request the request object containing parameters
+ * @param requestSupplier the supplier that creates the streaming API call
+ * @param tRespClass the class of the response type
+ * @param tDataClass the class of the data type in the stream
+ * @return the streaming response object containing the API result stream
+ */
+ public abstract , TResp extends FlowableClientResponse> TResp streamRequest(
+ TReq request, FlowableRequestSupplier> requestSupplier, Class tRespClass,
+ Class tDataClass);
+
+ /**
+ * Executes a Single API call synchronously and handles errors.
+ * @param the type of the response
+ * @param apiCall the Single API call to execute
+ * @return the response from the API call
+ * @throws ZAiHttpException if an HTTP error occurs with a parseable error body
+ * @throws HttpException if an HTTP error occurs that cannot be parsed
+ */
+ public static T execute(Single apiCall) {
+ try {
+ T response = apiCall.blockingGet();
+
+ // Check status code if the response is a Response type object
+ if (response instanceof Response) {
+ handleResponse((Response) response);
+ }
+
+ return response;
+ }
+ catch (HttpException e) {
+ logger.error("HTTP exception: {}", e.getMessage());
+ try {
+ if (e.response() == null || e.response().errorBody() == null) {
+ throw e;
+ }
+ String errorBody = e.response().errorBody().string();
+
+ ZAiError error = mapper.readValue(errorBody, ZAiError.class);
+
+ throw new ZAiHttpException(error, e, e.code());
+ }
+ catch (IOException ex) {
+ // couldn't parse ZAiError error
+ throw e;
+ }
+ }
+ }
+
+ /**
+ * Handles HTTP response and throws exception if not successful.
+ * @param response the HTTP response to handle
+ * @throws HttpException if the response is not successful
+ */
+ private static void handleResponse(Response response) {
+ if (!response.isSuccessful()) {
+ throw new HttpException(response);
+ }
+ }
+
+ /**
+ * Creates a streaming Flowable that maps SSE data to the specified class type.
+ * @param the type to map the SSE data to
+ * @param apiCall the API call that returns streaming data
+ * @param cl the class to map the SSE data to
+ * @return a Flowable of the specified type
+ */
+ public Flowable stream(retrofit2.Call apiCall, Class cl) {
+ return stream(apiCall).map(sse -> mapper.readValue(sse.getData(), cl));
+ }
+
+ /**
+ * Creates a streaming Flowable of SSE events without emitting done events.
+ * @param apiCall the API call that returns streaming data
+ * @return a Flowable of SSE events
+ */
+ public static Flowable stream(retrofit2.Call apiCall) {
+ return stream(apiCall, false);
+ }
+
+ /**
+ * Creates a streaming Flowable of SSE events with optional done event emission.
+ * @param apiCall the API call that returns streaming data
+ * @param emitDone whether to emit done events
+ * @return a Flowable of SSE events
+ */
+ public static Flowable stream(retrofit2.Call apiCall, boolean emitDone) {
+ return Flowable.create(emitter -> apiCall.enqueue(new ResponseBodyCallback(emitter, emitDone)),
+ BackpressureStrategy.BUFFER);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/CommonRequest.java b/core/src/main/java/ai/z/openapi/service/CommonRequest.java
new file mode 100644
index 0000000..71a703f
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/CommonRequest.java
@@ -0,0 +1,38 @@
+package ai.z.openapi.service;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.Map;
+
+/**
+ * Common request base class with extra JSON fields. Other ZAI request classes should
+ * extend this class to inherit common request parameters.
+ */
+
+@NoArgsConstructor
+@AllArgsConstructor
+@SuperBuilder
+@Data
+public class CommonRequest {
+
+ /**
+ * Request ID provided by the client, must be unique. Used to distinguish each
+ * request. If not provided by the client, the platform will generate one by default.
+ */
+ @JsonProperty("request_id")
+ private String requestId;
+
+ /**
+ * A unique identifier representing your end-user, which will help ZAI to monitor and
+ * detect abuse.
+ */
+ @JsonProperty("user_id")
+ private String userId;
+
+ private Map extraJson;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/agents/AgentAsyncResultRetrieveParams.java b/core/src/main/java/ai/z/openapi/service/agents/AgentAsyncResultRetrieveParams.java
new file mode 100644
index 0000000..5d80dc5
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/agents/AgentAsyncResultRetrieveParams.java
@@ -0,0 +1,43 @@
+package ai.z.openapi.service.agents;
+
+import ai.z.openapi.core.model.ClientRequest;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Parameters for retrieving agent asynchronous task results. This class contains the
+ * necessary parameters to query the result of an agent's async task.
+ */
+@EqualsAndHashCode(callSuper = false)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AgentAsyncResultRetrieveParams implements ClientRequest> {
+
+ /**
+ * The ID of the agent async task to retrieve
+ */
+ @JsonProperty("task_id")
+ private String taskId;
+
+ /**
+ * The agent ID associated with the async task
+ */
+ @JsonProperty("agent_id")
+ private String agentId;
+
+ /**
+ * Optional request ID for tracking
+ */
+ @JsonProperty("request_id")
+ private String requestId;
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/agents/AgentService.java b/core/src/main/java/ai/z/openapi/service/agents/AgentService.java
new file mode 100644
index 0000000..62e0a0a
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/agents/AgentService.java
@@ -0,0 +1,27 @@
+package ai.z.openapi.service.agents;
+
+import ai.z.openapi.service.model.ChatCompletionResponse;
+import ai.z.openapi.service.model.ModelData;
+import io.reactivex.Single;
+
+/**
+ * Agent completion service interface
+ */
+public interface AgentService {
+
+ /**
+ * Creates an agent completion, either streaming or non-streaming based on the request
+ * configuration.
+ * @param request the agents completion request
+ * @return ChatCompletionResponse containing the agent completion result
+ */
+ ChatCompletionResponse createAgentCompletion(AgentsCompletionRequest request);
+
+ /**
+ * Retrieves the result of an asynchronous agent operation.
+ * @param request the query request for the async agent result
+ * @return Single containing the async agent operation result
+ */
+ Single retrieveAgentAsyncResult(AgentAsyncResultRetrieveParams request);
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/agents/AgentServiceImpl.java b/core/src/main/java/ai/z/openapi/service/agents/AgentServiceImpl.java
new file mode 100644
index 0000000..6225497
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/agents/AgentServiceImpl.java
@@ -0,0 +1,52 @@
+package ai.z.openapi.service.agents;
+
+import ai.z.openapi.ZaiClient;
+import ai.z.openapi.api.agents.AgentsApi;
+import ai.z.openapi.service.model.ChatCompletionResponse;
+import ai.z.openapi.service.model.ModelData;
+import ai.z.openapi.utils.FlowableRequestSupplier;
+import ai.z.openapi.utils.RequestSupplier;
+import io.reactivex.Single;
+import okhttp3.ResponseBody;
+
+/**
+ * Agent completion service implementation
+ */
+public class AgentServiceImpl implements AgentService {
+
+ private final ZaiClient zAiClient;
+
+ private final AgentsApi agentsApi;
+
+ public AgentServiceImpl(ZaiClient zAiClient) {
+ this.zAiClient = zAiClient;
+ this.agentsApi = zAiClient.retrofit().create(AgentsApi.class);
+ }
+
+ @Override
+ public ChatCompletionResponse createAgentCompletion(AgentsCompletionRequest request) {
+ if (request.getStream()) {
+ return streamAgentCompletion(request);
+ }
+ else {
+ return syncAgentCompletion(request);
+ }
+ }
+
+ @Override
+ public Single retrieveAgentAsyncResult(AgentAsyncResultRetrieveParams request) {
+ return agentsApi.queryAgentsAsyncResult(request);
+ }
+
+ private ChatCompletionResponse streamAgentCompletion(AgentsCompletionRequest request) {
+ FlowableRequestSupplier> supplier = agentsApi::agentsCompletionStream;
+ ;
+ return this.zAiClient.streamRequest(request, supplier, ChatCompletionResponse.class, ModelData.class);
+ }
+
+ private ChatCompletionResponse syncAgentCompletion(AgentsCompletionRequest request) {
+ RequestSupplier supplier = agentsApi::agentsCompletionSync;
+ return this.zAiClient.executeRequest(request, supplier, ChatCompletionResponse.class);
+ }
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/agents/AgentsCompletionRequest.java b/core/src/main/java/ai/z/openapi/service/agents/AgentsCompletionRequest.java
new file mode 100644
index 0000000..69fe15a
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/agents/AgentsCompletionRequest.java
@@ -0,0 +1,60 @@
+package ai.z.openapi.service.agents;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import ai.z.openapi.service.model.ChatMessage;
+import ai.z.openapi.service.model.SensitiveWordCheckRequest;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Request parameters for agent completion API calls. This class contains all the
+ * necessary parameters to initiate an agent completion request, including agent ID,
+ * messages, streaming options, and custom variables.
+ */
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AgentsCompletionRequest extends CommonRequest implements ClientRequest {
+
+ /**
+ * Agent ID
+ */
+ @JsonProperty("agent_id")
+ private String agent_id;
+
+ /**
+ * Message body
+ */
+ private List messages;
+
+ /**
+ * Synchronous call: false, SSE call: true
+ */
+ private Boolean stream;
+
+ /**
+ * Sensitive word detection control
+ */
+ @JsonProperty("sensitive_word_check")
+ private SensitiveWordCheckRequest sensitiveWordCheck;
+
+ /**
+ * Agent business fields
+ * @return
+ */
+ @JsonProperty("custom_variables")
+ private ObjectNode custom_variables;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantApiResponse.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantApiResponse.java
new file mode 100644
index 0000000..a88e345
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantApiResponse.java
@@ -0,0 +1,62 @@
+package ai.z.openapi.service.assistant;
+
+import ai.z.openapi.core.model.FlowableClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import io.reactivex.Flowable;
+import lombok.Data;
+
+/**
+ * Response wrapper for Assistant API calls that supports both synchronous and streaming
+ * responses. This class implements FlowableClientResponse to handle streaming assistant
+ * completions.
+ */
+@Data
+public class AssistantApiResponse implements FlowableClientResponse {
+
+ /**
+ * Response status code.
+ */
+ private int code;
+
+ /**
+ * Response message.
+ */
+ private String msg;
+
+ /**
+ * Indicates whether the request was successful.
+ */
+ private boolean success;
+
+ /**
+ * The assistant completion data for synchronous responses.
+ */
+ private AssistantCompletion data;
+
+ /**
+ * The flowable stream for streaming responses.
+ */
+ private Flowable flowable;
+
+ /**
+ * Error information if the request failed.
+ */
+ private ChatError error;
+
+ /**
+ * Default constructor.
+ */
+ public AssistantApiResponse() {
+ }
+
+ /**
+ * Constructor with code and message.
+ * @param code the response code
+ * @param msg the response message
+ */
+ public AssistantApiResponse(int code, String msg) {
+ this.code = code;
+ this.msg = msg;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantAttachments.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantAttachments.java
new file mode 100644
index 0000000..95d3c6e
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantAttachments.java
@@ -0,0 +1,26 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * This class represents an attachment with a file ID.
+ */
+public class AssistantAttachments {
+
+ /**
+ * The ID of the file attachment.
+ */
+ @JsonProperty("file_id")
+ private String fileId;
+
+ // Getters and Setters
+
+ public String getFileId() {
+ return fileId;
+ }
+
+ public void setFileId(String fileId) {
+ this.fileId = fileId;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantChoice.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantChoice.java
new file mode 100644
index 0000000..d16c80e
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantChoice.java
@@ -0,0 +1,138 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.node.JsonNodeFactory;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.assistant.message.MessageContent;
+import ai.z.openapi.service.deserialize.MessageDeserializeFactory;
+import ai.z.openapi.service.deserialize.assistant.AssistantChoiceDeserializer;
+
+import java.util.Iterator;
+import java.util.Map;
+
+/**
+ * Represents an assistant's choice output in conversation responses. This class contains
+ * the response index, message content, finish reason, and metadata.
+ */
+@JsonDeserialize(using = AssistantChoiceDeserializer.class)
+public class AssistantChoice extends ObjectNode {
+
+ /**
+ * Result index.
+ */
+ @JsonProperty("index")
+ private int index;
+
+ /**
+ * Current conversation output message content.
+ */
+ @JsonProperty("delta")
+ private MessageContent delta;
+
+ /**
+ * Reason for inference completion: - stop: inference naturally ended or triggered
+ * stop words - sensitive: model inference content was blocked by security review -
+ * network_error: model inference service exception
+ */
+ @JsonProperty("finish_reason")
+ private String finishReason;
+
+ /**
+ * Metadata, extension field.
+ */
+ @JsonProperty("metadata")
+ private Map metadata;
+
+ public AssistantChoice() {
+ super(JsonNodeFactory.instance);
+ }
+
+ public AssistantChoice(ObjectNode objectNode) {
+ super(JsonNodeFactory.instance);
+
+ if (objectNode == null) {
+ return;
+ }
+
+ ObjectMapper objectMapper = MessageDeserializeFactory.defaultObjectMapper();
+ if (objectNode.has("index")) {
+ this.setIndex(objectNode.get("index").asInt());
+ }
+ else {
+ this.setIndex(0);
+ }
+ if (objectNode.has("delta")) {
+ MessageContent delta = objectMapper.convertValue(objectNode.get("delta"), MessageContent.class);
+ this.setDelta(delta);
+ }
+ else {
+ this.setDelta(null);
+ }
+ if (objectNode.has("finish_reason")) {
+ this.setFinishReason(objectNode.get("finish_reason").asText());
+
+ }
+ else {
+ this.setFinishReason(null);
+ }
+ if (objectNode.has("metadata")) {
+ Map metadata = objectMapper.convertValue(objectNode.get("metadata"),
+ new TypeReference>() {
+ });
+
+ this.setMetadata(metadata);
+ }
+ else {
+ this.setMetadata(null);
+ }
+
+ Iterator fieldNames = objectNode.fieldNames();
+ while (fieldNames.hasNext()) {
+ String fieldName = fieldNames.next();
+ JsonNode field = objectNode.get(fieldName);
+ this.set(fieldName, field);
+ }
+ }
+ // Getters and Setters
+
+ public int getIndex() {
+ return index;
+ }
+
+ public void setIndex(int index) {
+ this.index = index;
+ this.put("index", index);
+ }
+
+ public MessageContent getDelta() {
+ return delta;
+ }
+
+ public void setDelta(MessageContent delta) {
+ this.delta = delta;
+ this.putPOJO("delta", delta);
+ }
+
+ public String getFinishReason() {
+ return finishReason;
+ }
+
+ public void setFinishReason(String finishReason) {
+ this.finishReason = finishReason;
+ this.put("finish_reason", finishReason);
+ }
+
+ public Map getMetadata() {
+ return metadata;
+ }
+
+ public void setMetadata(Map metadata) {
+ this.metadata = metadata;
+ this.putPOJO("metadata", metadata);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantCompletion.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantCompletion.java
new file mode 100644
index 0000000..65b37c5
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantCompletion.java
@@ -0,0 +1,254 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.JsonNodeFactory;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.MessageDeserializeFactory;
+import ai.z.openapi.service.deserialize.assistant.AssistantCompletionDeserializer;
+
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * This class represents the completion data returned by an assistant.
+ */
+@JsonDeserialize(using = AssistantCompletionDeserializer.class)
+public class AssistantCompletion extends ObjectNode {
+
+ /**
+ * Request ID
+ */
+ @JsonProperty("id")
+ private String id;
+
+ /**
+ * Conversation ID
+ */
+ @JsonProperty("conversation_id")
+ private String conversationId;
+
+ /**
+ * Assistant ID
+ */
+ @JsonProperty("assistant_id")
+ private String assistantId;
+
+ /**
+ * Request creation time, Unix timestamp
+ */
+ @JsonProperty("created")
+ private int created;
+
+ /**
+ * Return status, including: `completed` indicates generation finished, `in_progress`
+ * indicates generating, `failed` indicates generation exception
+ */
+ @JsonProperty("status")
+ private String status;
+
+ /**
+ * Error information
+ */
+ @JsonProperty("last_error")
+ private ErrorInfo lastError;
+
+ /**
+ * Incremental return information
+ */
+ @JsonProperty("choices")
+ private List choices;
+
+ /**
+ * Metadata, extension field
+ */
+ @JsonProperty("metadata")
+ private Map metadata;
+
+ /**
+ * Token count statistics
+ */
+ @JsonProperty("usage")
+ private CompletionUsage usage;
+
+ public AssistantCompletion() {
+ super(JsonNodeFactory.instance);
+ }
+
+ public AssistantCompletion(ObjectNode objectNode) {
+ super(JsonNodeFactory.instance);
+
+ ObjectMapper objectMapper = MessageDeserializeFactory.defaultObjectMapper();
+ if (objectNode.has("id")) {
+ this.setId(objectNode.get("id").asText());
+ }
+ else {
+ this.setId(null);
+ }
+
+ if (objectNode.has("conversation_id")) {
+ this.setConversationId(objectNode.get("conversation_id").asText());
+ }
+ else {
+ this.setConversationId(null);
+ }
+
+ if (objectNode.has("assistant_id")) {
+ this.setAssistantId(objectNode.get("assistant_id").asText());
+ }
+ else {
+ this.setAssistantId(null);
+ }
+
+ if (objectNode.has("created")) {
+ this.setCreated(objectNode.get("created").asInt());
+ }
+ else {
+ this.setCreated(0);
+ }
+
+ if (objectNode.has("status")) {
+ this.setStatus(objectNode.get("status").asText());
+
+ }
+ else {
+ this.setStatus(null);
+ }
+
+ if (objectNode.has("last_error")) {
+ this.setLastError(objectMapper.convertValue(objectNode.get("last_error"), ErrorInfo.class));
+ }
+ else {
+ this.setLastError(null);
+ }
+
+ if (objectNode.has("choices")) {
+ List choices1 = objectMapper.convertValue(objectNode.get("choices"),
+ new com.fasterxml.jackson.core.type.TypeReference>() {
+ });
+
+ this.setChoices(choices1);
+
+ }
+ else {
+ this.setChoices(null);
+ }
+
+ if (objectNode.has("metadata")) {
+ this.setMetadata(objectMapper.convertValue(objectNode.get("metadata"), Map.class));
+ }
+ else {
+ this.setMetadata(null);
+ }
+
+ if (objectNode.has("usage")) {
+ this.setUsage(objectMapper.convertValue(objectNode.get("usage"), CompletionUsage.class));
+ }
+ else {
+ this.setUsage(null);
+ }
+
+ Iterator fieldNames = objectNode.fieldNames();
+ while (fieldNames.hasNext()) {
+ String fieldName = fieldNames.next();
+ JsonNode field = objectNode.get(fieldName);
+ this.set(fieldName, field);
+ }
+ }
+ // Getters and Setters
+
+ public String getId() {
+ return id;
+ }
+
+ public void setId(String id) {
+ this.id = id;
+ this.put("id", id);
+ }
+
+ public String getConversationId() {
+ return conversationId;
+ }
+
+ public void setConversationId(String conversationId) {
+ this.conversationId = conversationId;
+ this.put("conversation_id", conversationId);
+ }
+
+ public String getAssistantId() {
+ return assistantId;
+ }
+
+ public void setAssistantId(String assistantId) {
+ this.assistantId = assistantId;
+ this.put("assistant_id", assistantId);
+ }
+
+ public int getCreated() {
+ return created;
+ }
+
+ public void setCreated(int created) {
+ this.created = created;
+ this.put("created", created);
+ }
+
+ public String getStatus() {
+ return status;
+ }
+
+ public void setStatus(String status) {
+ this.status = status;
+ this.put("status", status);
+ }
+
+ public ErrorInfo getLastError() {
+ return lastError;
+ }
+
+ public void setLastError(ErrorInfo lastError) {
+ this.lastError = lastError;
+ this.putPOJO("last_error", lastError);
+ }
+
+ public List getChoices() {
+ return choices;
+ }
+
+ public void setChoices(List choices) {
+ this.choices = choices;
+ ArrayNode jsonNodes = this.putArray("choices");
+ if (choices == null) {
+ jsonNodes.removeAll();
+ }
+ else {
+
+ for (AssistantChoice choice : choices) {
+ jsonNodes.add(choice);
+ }
+ }
+ }
+
+ public Map getMetadata() {
+ return metadata;
+ }
+
+ public void setMetadata(Map metadata) {
+ this.metadata = metadata;
+ this.putPOJO("metadata", metadata);
+ }
+
+ public CompletionUsage getUsage() {
+ return usage;
+ }
+
+ public void setUsage(CompletionUsage usage) {
+ this.usage = usage;
+ this.putPOJO("usage", usage);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantExtraParameters.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantExtraParameters.java
new file mode 100644
index 0000000..28c4a1c
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantExtraParameters.java
@@ -0,0 +1,23 @@
+package ai.z.openapi.service.assistant;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+/**
+ * Extra parameters for assistant configuration. This class contains additional optional
+ * parameters that can be used to customize assistant behavior and functionality.
+ */
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AssistantExtraParameters {
+
+ /**
+ * Translation agent parameters
+ */
+ private TranslateParameters translate;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantParameters.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantParameters.java
new file mode 100644
index 0000000..a0c07da
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantParameters.java
@@ -0,0 +1,71 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * This class represents the parameters for an assistant, including optional fields.
+ */
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AssistantParameters extends CommonRequest implements ClientRequest {
+
+ /**
+ * The ID of the assistant.
+ */
+ @JsonProperty("assistant_id")
+ private String assistantId;
+
+ /**
+ * The conversation ID. If not provided, a new conversation is created.
+ */
+ @JsonProperty("conversation_id")
+ private String conversationId;
+
+ /**
+ * The name of the model, default is 'GLM-4-Assistant'.
+ */
+ @JsonProperty("model")
+ private String model;
+
+ /**
+ * Whether to support streaming SSE, should be set to True.
+ */
+ @JsonProperty("stream")
+ private boolean stream;
+
+ /**
+ * The list of conversation messages.
+ */
+ @JsonProperty("messages")
+ private List messages;
+
+ /**
+ * The list of file attachments for the conversation, optional.
+ */
+ @JsonProperty("attachments")
+ private List attachments;
+
+ /**
+ * Metadata or additional fields, optional.
+ */
+ @JsonProperty("metadata")
+ private Map metadata;
+
+ @JsonProperty("extra_parameters")
+ private AssistantExtraParameters extraParameters;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantService.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantService.java
new file mode 100644
index 0000000..b6f81d7
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantService.java
@@ -0,0 +1,43 @@
+package ai.z.openapi.service.assistant;
+
+import ai.z.openapi.service.assistant.AssistantParameters;
+import ai.z.openapi.service.assistant.AssistantApiResponse;
+import ai.z.openapi.service.assistant.conversation.ConversationParameters;
+import ai.z.openapi.service.assistant.conversation.ConversationUsageListResponse;
+import ai.z.openapi.service.assistant.query_support.AssistantSupportResponse;
+import ai.z.openapi.service.assistant.query_support.QuerySupportParams;
+
+/**
+ * Assistant service interface
+ */
+public interface AssistantService {
+
+ /**
+ * Creates a streaming assistant completion.
+ * @param request the assistant completion request
+ * @return AssistantApiResponse containing the completion result
+ */
+ AssistantApiResponse assistantCompletionStream(AssistantParameters request);
+
+ /**
+ * Creates a non-streaming assistant completion.
+ * @param request the assistant completion request
+ * @return AssistantApiResponse containing the completion result
+ */
+ AssistantApiResponse assistantCompletion(AssistantParameters request);
+
+ /**
+ * Queries assistant support status.
+ * @param request the query support request
+ * @return AssistantSupportResponse containing the support information
+ */
+ AssistantSupportResponse querySupport(QuerySupportParams request);
+
+ /**
+ * Queries conversation usage information.
+ * @param request the conversation parameters
+ * @return ConversationUsageListResponse containing the usage information
+ */
+ ConversationUsageListResponse queryConversationUsage(ConversationParameters request);
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/AssistantServiceImpl.java b/core/src/main/java/ai/z/openapi/service/assistant/AssistantServiceImpl.java
new file mode 100644
index 0000000..2a36689
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/AssistantServiceImpl.java
@@ -0,0 +1,53 @@
+package ai.z.openapi.service.assistant;
+
+import ai.z.openapi.ZaiClient;
+import ai.z.openapi.api.assistant.AssistantApi;
+import ai.z.openapi.service.assistant.conversation.ConversationParameters;
+import ai.z.openapi.service.assistant.conversation.ConversationUsageListResponse;
+import ai.z.openapi.service.assistant.conversation.ConversationUsageListStatus;
+import ai.z.openapi.service.assistant.query_support.AssistantSupportResponse;
+import ai.z.openapi.service.assistant.query_support.AssistantSupportStatus;
+import ai.z.openapi.service.assistant.query_support.QuerySupportParams;
+import ai.z.openapi.utils.FlowableRequestSupplier;
+import ai.z.openapi.utils.RequestSupplier;
+import okhttp3.ResponseBody;
+
+/**
+ * Implementation of AssistantService
+ */
+public class AssistantServiceImpl implements AssistantService {
+
+ private final ZaiClient zAiClient;
+
+ private final AssistantApi assistantApi;
+
+ public AssistantServiceImpl(ZaiClient zAiClient) {
+ this.zAiClient = zAiClient;
+ this.assistantApi = zAiClient.retrofit().create(AssistantApi.class);
+ }
+
+ @Override
+ public AssistantApiResponse assistantCompletionStream(AssistantParameters request) {
+ FlowableRequestSupplier> supplier = assistantApi::assistantCompletionStream;
+ return zAiClient.streamRequest(request, supplier, AssistantApiResponse.class, AssistantCompletion.class);
+ }
+
+ @Override
+ public AssistantApiResponse assistantCompletion(AssistantParameters request) {
+ RequestSupplier supplier = assistantApi::assistantCompletion;
+ return zAiClient.executeRequest(request, supplier, AssistantApiResponse.class);
+ }
+
+ @Override
+ public AssistantSupportResponse querySupport(QuerySupportParams request) {
+ RequestSupplier supplier = assistantApi::querySupport;
+ return zAiClient.executeRequest(request, supplier, AssistantSupportResponse.class);
+ }
+
+ @Override
+ public ConversationUsageListResponse queryConversationUsage(ConversationParameters request) {
+ RequestSupplier supplier = assistantApi::queryConversationUsage;
+ return zAiClient.executeRequest(request, supplier, ConversationUsageListResponse.class);
+ }
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/CompletionUsage.java b/core/src/main/java/ai/z/openapi/service/assistant/CompletionUsage.java
new file mode 100644
index 0000000..25c5006
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/CompletionUsage.java
@@ -0,0 +1,101 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.node.JsonNodeFactory;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.MessageDeserializeFactory;
+import ai.z.openapi.service.deserialize.assistant.CompletionUsageDeserializer;
+
+import java.util.Iterator;
+
+/**
+ * This class represents the usage statistics for a completion.
+ */
+@JsonDeserialize(using = CompletionUsageDeserializer.class)
+public class CompletionUsage extends ObjectNode {
+
+ /**
+ * Number of tokens in the input (prompt).
+ */
+ @JsonProperty("prompt_tokens")
+ private int promptTokens;
+
+ /**
+ * Number of tokens in the output (completion).
+ */
+ @JsonProperty("completion_tokens")
+ private int completionTokens;
+
+ /**
+ * Total number of tokens used.
+ */
+ @JsonProperty("total_tokens")
+ private int totalTokens;
+
+ public CompletionUsage() {
+ super(JsonNodeFactory.instance);
+ }
+
+ public CompletionUsage(ObjectNode objectNode) {
+ super(JsonNodeFactory.instance);
+
+ ObjectMapper objectMapper = MessageDeserializeFactory.defaultObjectMapper();
+ if (objectNode.get("prompt_tokens") != null) {
+ this.setPromptTokens(objectNode.get("prompt_tokens").asInt());
+ }
+ else {
+ this.setPromptTokens(0);
+ }
+ if (objectNode.get("completion_tokens") != null) {
+ this.setCompletionTokens(objectNode.get("completion_tokens").asInt());
+ }
+ else {
+ this.setCompletionTokens(0);
+ }
+ if (objectNode.get("total_tokens") != null) {
+ this.setTotalTokens(objectNode.get("total_tokens").asInt());
+ }
+ else {
+ this.setTotalTokens(0);
+ }
+
+ Iterator fieldNames = objectNode.fieldNames();
+ while (fieldNames.hasNext()) {
+ String fieldName = fieldNames.next();
+ JsonNode field = objectNode.get(fieldName);
+ this.set(fieldName, field);
+ }
+ }
+ // Getters and Setters
+
+ public int getPromptTokens() {
+ return promptTokens;
+ }
+
+ public void setPromptTokens(int promptTokens) {
+ this.promptTokens = promptTokens;
+ this.put("prompt_tokens", promptTokens);
+ }
+
+ public int getCompletionTokens() {
+ return completionTokens;
+ }
+
+ public void setCompletionTokens(int completionTokens) {
+ this.completionTokens = completionTokens;
+ this.put("completion_tokens", completionTokens);
+ }
+
+ public int getTotalTokens() {
+ return totalTokens;
+ }
+
+ public void setTotalTokens(int totalTokens) {
+ this.totalTokens = totalTokens;
+ this.put("total_tokens", totalTokens);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/ConversationMessage.java b/core/src/main/java/ai/z/openapi/service/assistant/ConversationMessage.java
new file mode 100644
index 0000000..936ce34
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/ConversationMessage.java
@@ -0,0 +1,34 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.List;
+
+/**
+ * This class represents a conversation message body.
+ */
+@EqualsAndHashCode(callSuper = false)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class ConversationMessage {
+
+ /**
+ * The role of the user input, e.g., 'user'.
+ */
+ @JsonProperty("role")
+ private String role;
+
+ /**
+ * The content of the conversation message.
+ */
+ @JsonProperty("content")
+ private List content;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/ErrorInfo.java b/core/src/main/java/ai/z/openapi/service/assistant/ErrorInfo.java
new file mode 100644
index 0000000..ae7fad4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/ErrorInfo.java
@@ -0,0 +1,80 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.node.JsonNodeFactory;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.MessageDeserializeFactory;
+import ai.z.openapi.service.deserialize.assistant.ErrorInfoDeserializer;
+
+import java.util.Iterator;
+
+/**
+ * This class represents error information.
+ */
+@JsonDeserialize(using = ErrorInfoDeserializer.class)
+public class ErrorInfo extends ObjectNode {
+
+ /**
+ * Error code.
+ */
+ @JsonProperty("code")
+ private String code;
+
+ /**
+ * Error message.
+ */
+ @JsonProperty("message")
+ private String message;
+
+ public ErrorInfo() {
+ super(JsonNodeFactory.instance);
+ }
+
+ public ErrorInfo(ObjectNode objectNode) {
+ super(JsonNodeFactory.instance);
+
+ ObjectMapper objectMapper = MessageDeserializeFactory.defaultObjectMapper();
+
+ if (objectNode.get("code") != null) {
+ this.setCode(objectNode.get("code").asText());
+ }
+ else {
+ this.setCode(null);
+ }
+ if (objectNode.get("message") != null) {
+ this.setMessage(objectNode.get("message").asText());
+ }
+ else {
+ this.setMessage(null);
+ }
+ Iterator fieldNames = objectNode.fieldNames();
+ while (fieldNames.hasNext()) {
+ String fieldName = fieldNames.next();
+ JsonNode field = objectNode.get(fieldName);
+ this.set(fieldName, field);
+ }
+ }
+
+ // Getters and Setters
+ public String getCode() {
+ return code;
+ }
+
+ public void setCode(String code) {
+ this.code = code;
+ this.put("code", code);
+ }
+
+ public String getMessage() {
+ return message;
+ }
+
+ public void setMessage(String message) {
+ this.message = message;
+ this.put("message", message);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/MessageTextContent.java b/core/src/main/java/ai/z/openapi/service/assistant/MessageTextContent.java
new file mode 100644
index 0000000..70ad537
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/MessageTextContent.java
@@ -0,0 +1,33 @@
+package ai.z.openapi.service.assistant;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+/**
+ * This class represents the text content of a message. Currently supports only type =
+ * text.
+ */
+@EqualsAndHashCode(callSuper = false)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class MessageTextContent {
+
+ /**
+ * The type of the message content, currently only "text" is supported.
+ */
+ @JsonProperty("type")
+ private String type;
+
+ /**
+ * The text content of the message.
+ */
+ @JsonProperty("text")
+ private String text;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/TranslateParameters.java b/core/src/main/java/ai/z/openapi/service/assistant/TranslateParameters.java
new file mode 100644
index 0000000..21a840f
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/TranslateParameters.java
@@ -0,0 +1,28 @@
+package ai.z.openapi.service.assistant;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+/**
+ * Parameters for translation operations. This class contains the source and target
+ * language settings for translation requests.
+ */
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class TranslateParameters {
+
+ /**
+ * Source language for translation.
+ */
+ private String from;
+
+ /**
+ * Target language for translation.
+ */
+ private String to;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationParameters.java b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationParameters.java
new file mode 100644
index 0000000..dd8d99d
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationParameters.java
@@ -0,0 +1,40 @@
+package ai.z.openapi.service.assistant.conversation;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+/**
+ * This class represents the parameters for a conversation, including pagination.
+ */
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class ConversationParameters extends CommonRequest implements ClientRequest {
+
+ /**
+ * The Assistant ID.
+ */
+ @JsonProperty("assistant_id")
+ private String assistantId;
+
+ /**
+ * The current page number for pagination.
+ */
+ @JsonProperty("page")
+ private int page;
+
+ /**
+ * The number of items per page for pagination.
+ */
+ @JsonProperty("page_size")
+ private int pageSize;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsage.java b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsage.java
new file mode 100644
index 0000000..6a2fff5
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsage.java
@@ -0,0 +1,44 @@
+package ai.z.openapi.service.assistant.conversation;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+/**
+ * This class represents the usage data for a specific conversation.
+ */
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class ConversationUsage {
+
+ /**
+ * The conversation ID.
+ */
+ @JsonProperty("id")
+ private String id;
+
+ /**
+ * The Assistant ID.
+ */
+ @JsonProperty("assistant_id")
+ private String assistantId;
+
+ /**
+ * The creation time of the conversation.
+ */
+ @JsonProperty("create_time")
+ private String createTime;
+
+ /**
+ * The last update time of the conversation.
+ */
+ @JsonProperty("update_time")
+ private String updateTime;
+
+ /**
+ * The usage statistics for the conversation.
+ */
+ @JsonProperty("usage")
+ private Usage usage;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageList.java b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageList.java
new file mode 100644
index 0000000..42aecd4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageList.java
@@ -0,0 +1,34 @@
+package ai.z.openapi.service.assistant.conversation;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * This class represents a list of conversation usage data.
+ */
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class ConversationUsageList {
+
+ /**
+ * The Assistant ID.
+ */
+ @JsonProperty("assistant_id")
+ private String assistantId;
+
+ /**
+ * Whether there are more pages of results.
+ */
+ @JsonProperty("has_more")
+ private boolean hasMore;
+
+ /**
+ * The list of conversation usage data.
+ */
+ @JsonProperty("conversation_list")
+ private List conversationList;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageListResponse.java b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageListResponse.java
new file mode 100644
index 0000000..faeac90
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageListResponse.java
@@ -0,0 +1,47 @@
+package ai.z.openapi.service.assistant.conversation;
+
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import lombok.Data;
+
+/**
+ * Response for conversation usage list API calls. This class contains the response data
+ * for retrieving conversation usage statistics.
+ */
+@Data
+public class ConversationUsageListResponse implements ClientResponse {
+
+ /**
+ * Response status code.
+ */
+ private int code;
+
+ /**
+ * Response message.
+ */
+ private String msg;
+
+ /**
+ * Indicates if the request was successful.
+ */
+ private boolean success;
+
+ /**
+ * The conversation usage list data.
+ */
+ private ConversationUsageListStatus data;
+
+ /**
+ * Error information if the request failed.
+ */
+ private ChatError error;
+
+ public ConversationUsageListResponse() {
+ }
+
+ public ConversationUsageListResponse(int code, String msg) {
+ this.code = code;
+ this.msg = msg;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageListStatus.java b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageListStatus.java
new file mode 100644
index 0000000..5454a15
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/conversation/ConversationUsageListStatus.java
@@ -0,0 +1,32 @@
+package ai.z.openapi.service.assistant.conversation;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+/**
+ * This class represents the response containing a list of conversation usage data.
+ */
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class ConversationUsageListStatus {
+
+ /**
+ * The response code.
+ */
+ @JsonProperty("code")
+ private int code;
+
+ /**
+ * The response message.
+ */
+ @JsonProperty("msg")
+ private String msg;
+
+ /**
+ * The data containing the conversation usage list.
+ */
+ @JsonProperty("data")
+ private ConversationUsageList data;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/conversation/Usage.java b/core/src/main/java/ai/z/openapi/service/assistant/conversation/Usage.java
new file mode 100644
index 0000000..ac76183
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/conversation/Usage.java
@@ -0,0 +1,32 @@
+package ai.z.openapi.service.assistant.conversation;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+/**
+ * This class represents the usage statistics for a conversation.
+ */
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class Usage {
+
+ /**
+ * The number of tokens in the user's input.
+ */
+ @JsonProperty("prompt_tokens")
+ private int promptTokens;
+
+ /**
+ * The number of tokens in the model's input.
+ */
+ @JsonProperty("completion_tokens")
+ private int completionTokens;
+
+ /**
+ * The total number of tokens.
+ */
+ @JsonProperty("total_tokens")
+ private int totalTokens;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/MessageContent.java b/core/src/main/java/ai/z/openapi/service/assistant/message/MessageContent.java
new file mode 100644
index 0000000..cf1c3dc
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/MessageContent.java
@@ -0,0 +1,13 @@
+package ai.z.openapi.service.assistant.message;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import ai.z.openapi.service.deserialize.JsonTypeMapping;
+import ai.z.openapi.service.deserialize.assistant.message.MessageContentDeserializer;
+
+@JsonTypeMapping({ ToolsDeltaBlock.class, TextContentBlock.class })
+@JsonDeserialize(using = MessageContentDeserializer.class)
+@JsonIgnoreProperties(ignoreUnknown = true)
+public abstract class MessageContent {
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/TextContentBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/TextContentBlock.java
new file mode 100644
index 0000000..7d4b9a4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/TextContentBlock.java
@@ -0,0 +1,44 @@
+package ai.z.openapi.service.assistant.message;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+/**
+ * This class represents a block of text content in a conversation.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("content")
+public class TextContentBlock extends MessageContent {
+
+ /**
+ * The content of the text block.
+ */
+ @JsonProperty("content")
+ private String content;
+
+ /**
+ * The role of the speaker, default is "assistant".
+ */
+ @JsonProperty("role")
+ private String role = "assistant";
+
+ // Getters and Setters
+
+ public String getContent() {
+ return content;
+ }
+
+ public void setContent(String content) {
+ this.content = content;
+ }
+
+ public String getRole() {
+ return role;
+ }
+
+ public void setRole(String role) {
+ this.role = role;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/ToolsDeltaBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/ToolsDeltaBlock.java
new file mode 100644
index 0000000..5e2bdf6
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/ToolsDeltaBlock.java
@@ -0,0 +1,47 @@
+package ai.z.openapi.service.assistant.message;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+import java.util.List;
+
+/**
+ * This class represents a block of tool call data in a conversation.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("tool_calls")
+public class ToolsDeltaBlock extends MessageContent {
+
+ /**
+ * A list of tool call types.
+ */
+ @JsonProperty("tool_calls")
+ private List toolCalls;
+
+ /**
+ * The role of the speaker, default is "tool".
+ */
+ @JsonProperty("role")
+ private String role = "tool";
+
+ // Getters and Setters
+
+ public List getToolCalls() {
+ return toolCalls;
+ }
+
+ public void setToolCalls(List toolCalls) {
+ this.toolCalls = toolCalls;
+ }
+
+ public String getRole() {
+ return role;
+ }
+
+ public void setRole(String role) {
+ this.role = role;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/ToolsType.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/ToolsType.java
new file mode 100644
index 0000000..40b4eab
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/ToolsType.java
@@ -0,0 +1,19 @@
+package ai.z.openapi.service.assistant.message.tools;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import ai.z.openapi.service.assistant.message.tools.code_interpreter.CodeInterpreterToolBlock;
+import ai.z.openapi.service.assistant.message.tools.drawing_tool.DrawingToolBlock;
+import ai.z.openapi.service.assistant.message.tools.function.FunctionToolBlock;
+import ai.z.openapi.service.assistant.message.tools.retrieval.RetrievalToolBlock;
+import ai.z.openapi.service.assistant.message.tools.web_browser.WebBrowserToolBlock;
+import ai.z.openapi.service.deserialize.JsonTypeMapping;
+import ai.z.openapi.service.deserialize.assistant.message.tools.ToolsTypeDeserializer;
+
+@JsonTypeMapping({ WebBrowserToolBlock.class, RetrievalToolBlock.class, FunctionToolBlock.class, DrawingToolBlock.class,
+ CodeInterpreterToolBlock.class, })
+@JsonDeserialize(using = ToolsTypeDeserializer.class)
+@JsonIgnoreProperties(ignoreUnknown = true)
+public abstract class ToolsType {
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreter.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreter.java
new file mode 100644
index 0000000..b7ac8cd
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreter.java
@@ -0,0 +1,41 @@
+package ai.z.openapi.service.assistant.message.tools.code_interpreter;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.List;
+
+/**
+ * This class represents a code interpreter that executes code and returns the results.
+ */
+public class CodeInterpreter {
+
+ /**
+ * The generated code snippet that is input to the code sandbox.
+ */
+ @JsonProperty("input")
+ private String input;
+
+ /**
+ * The output results after the code execution.
+ */
+ @JsonProperty("outputs")
+ private List outputs;
+
+ // Getters and Setters
+
+ public String getInput() {
+ return input;
+ }
+
+ public void setInput(String input) {
+ this.input = input;
+ }
+
+ public List getOutputs() {
+ return outputs;
+ }
+
+ public void setOutputs(List outputs) {
+ this.outputs = outputs;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreterToolBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreterToolBlock.java
new file mode 100644
index 0000000..5d9327f
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreterToolBlock.java
@@ -0,0 +1,45 @@
+package ai.z.openapi.service.assistant.message.tools.code_interpreter;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+/**
+ * This class represents a block of code tool data.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("code_interpreter")
+public class CodeInterpreterToolBlock extends ToolsType {
+
+ /**
+ * The code interpreter object.
+ */
+ @JsonProperty("code_interpreter")
+ private CodeInterpreter codeInterpreter;
+
+ /**
+ * The type of tool being called, always "code_interpreter".
+ */
+ @JsonProperty("type")
+ private String type = "code_interpreter";
+
+ // Getters and Setters
+
+ public CodeInterpreter getCodeInterpreter() {
+ return codeInterpreter;
+ }
+
+ public void setCodeInterpreter(CodeInterpreter codeInterpreter) {
+ this.codeInterpreter = codeInterpreter;
+ }
+
+ public String getType() {
+ return type;
+ }
+
+ public void setType(String type) {
+ this.type = type;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreterToolOutput.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreterToolOutput.java
new file mode 100644
index 0000000..38152f0
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/code_interpreter/CodeInterpreterToolOutput.java
@@ -0,0 +1,54 @@
+package ai.z.openapi.service.assistant.message.tools.code_interpreter;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * This class represents the output result of a code tool.
+ */
+public class CodeInterpreterToolOutput {
+
+ /**
+ * The type of output, currently only "logs".
+ */
+ @JsonProperty("type")
+ private String type;
+
+ /**
+ * The log results from the code execution.
+ */
+ @JsonProperty("logs")
+ private String logs;
+
+ /**
+ * Error message if any occurred during code execution.
+ */
+ @JsonProperty("error_msg")
+ private String errorMsg;
+
+ // Getters and Setters
+
+ public String getType() {
+ return type;
+ }
+
+ public void setType(String type) {
+ this.type = type;
+ }
+
+ public String getLogs() {
+ return logs;
+ }
+
+ public void setLogs(String logs) {
+ this.logs = logs;
+ }
+
+ public String getErrorMsg() {
+ return errorMsg;
+ }
+
+ public void setErrorMsg(String errorMsg) {
+ this.errorMsg = errorMsg;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingTool.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingTool.java
new file mode 100644
index 0000000..a18ab67
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingTool.java
@@ -0,0 +1,41 @@
+package ai.z.openapi.service.assistant.message.tools.drawing_tool;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.List;
+
+/**
+ * The input string that was used to generate the drawing.
+ */
+public class DrawingTool {
+
+ /**
+ * The input string that was used to generate the drawing.
+ */
+ @JsonProperty("input")
+ private String input;
+
+ /**
+ * A list of outputs generated by the drawing tool.
+ */
+ @JsonProperty("outputs")
+ private List outputs;
+
+ // Getters and Setters
+
+ public String getInput() {
+ return input;
+ }
+
+ public void setInput(String input) {
+ this.input = input;
+ }
+
+ public List getOutputs() {
+ return outputs;
+ }
+
+ public void setOutputs(List outputs) {
+ this.outputs = outputs;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingToolBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingToolBlock.java
new file mode 100644
index 0000000..9a0d536
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingToolBlock.java
@@ -0,0 +1,45 @@
+package ai.z.openapi.service.assistant.message.tools.drawing_tool;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+/**
+ * This class represents a block of drawing tool data.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("drawing_tool")
+public class DrawingToolBlock extends ToolsType {
+
+ /**
+ * The drawing tool object that contains input and outputs.
+ */
+ @JsonProperty("drawing_tool")
+ private DrawingTool drawingTool;
+
+ /**
+ * The type of tool being called, always "drawing_tool".
+ */
+ @JsonProperty("type")
+ private String type = "drawing_tool";
+
+ // Getters and Setters
+
+ public DrawingTool getDrawingTool() {
+ return drawingTool;
+ }
+
+ public void setDrawingTool(DrawingTool drawingTool) {
+ this.drawingTool = drawingTool;
+ }
+
+ public String getType() {
+ return type;
+ }
+
+ public void setType(String type) {
+ this.type = type;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingToolOutput.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingToolOutput.java
new file mode 100644
index 0000000..5d060a0
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/drawing_tool/DrawingToolOutput.java
@@ -0,0 +1,26 @@
+package ai.z.openapi.service.assistant.message.tools.drawing_tool;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * This class represents the output of a drawing tool, containing the generated image.
+ */
+public class DrawingToolOutput {
+
+ /**
+ * The generated image in a string format.
+ */
+ @JsonProperty("image")
+ private String image;
+
+ // Getters and Setters
+
+ public String getImage() {
+ return image;
+ }
+
+ public void setImage(String image) {
+ this.image = image;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionTool.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionTool.java
new file mode 100644
index 0000000..810bc20
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionTool.java
@@ -0,0 +1,58 @@
+package ai.z.openapi.service.assistant.message.tools.function;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.JsonNode;
+
+import java.util.List;
+
+/**
+ * This class represents a function tool with a name, arguments, and outputs.
+ */
+public class FunctionTool {
+
+ /**
+ * The name of the function tool.
+ */
+ @JsonProperty("name")
+ private String name;
+
+ /**
+ * The arguments for the function tool, which can be a string or a dictionary.
+ */
+ @JsonProperty("arguments")
+ private JsonNode arguments; // Union type in Java can be represented by Object, and
+ // deserialization handles it accordingly
+
+ /**
+ * A list of outputs generated by the function tool.
+ */
+ @JsonProperty("outputs")
+ private List outputs;
+
+ // Getters and Setters
+
+ public String getName() {
+ return name;
+ }
+
+ public void setName(String name) {
+ this.name = name;
+ }
+
+ public JsonNode getArguments() {
+ return arguments;
+ }
+
+ public void setArguments(JsonNode arguments) {
+ this.arguments = arguments;
+ }
+
+ public List getOutputs() {
+ return outputs;
+ }
+
+ public void setOutputs(List outputs) {
+ this.outputs = outputs;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionToolBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionToolBlock.java
new file mode 100644
index 0000000..5345ef9
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionToolBlock.java
@@ -0,0 +1,45 @@
+package ai.z.openapi.service.assistant.message.tools.function;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+/**
+ * This class represents a block of function tool data.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("function")
+public class FunctionToolBlock extends ToolsType {
+
+ /**
+ * The function tool object that contains the name, arguments, and outputs.
+ */
+ @JsonProperty("function")
+ private FunctionTool function;
+
+ /**
+ * The type of tool being called, always "function".
+ */
+ @JsonProperty("type")
+ private String type = "function";
+
+ // Getters and Setters
+
+ public FunctionTool getFunction() {
+ return function;
+ }
+
+ public void setFunction(FunctionTool function) {
+ this.function = function;
+ }
+
+ public String getType() {
+ return type;
+ }
+
+ public void setType(String type) {
+ this.type = type;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionToolOutput.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionToolOutput.java
new file mode 100644
index 0000000..b2ab2ab
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/function/FunctionToolOutput.java
@@ -0,0 +1,24 @@
+package ai.z.openapi.service.assistant.message.tools.function;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * This class represents the output of a function tool, containing the generated content.
+ */
+public class FunctionToolOutput {
+
+ /**
+ * The generated content as a string.
+ */
+ @JsonProperty("content")
+ private String content;
+
+ public String getContent() {
+ return content;
+ }
+
+ public void setContent(String content) {
+ this.content = content;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalTool.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalTool.java
new file mode 100644
index 0000000..b14fb42
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalTool.java
@@ -0,0 +1,26 @@
+package ai.z.openapi.service.assistant.message.tools.retrieval;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.List;
+
+/**
+ * This class represents the outputs of a retrieval tool.
+ */
+public class RetrievalTool {
+
+ /**
+ * A list of text snippets and their respective document names retrieved from the
+ * knowledge base.
+ */
+ @JsonProperty("outputs")
+ private List outputs;
+
+ public List getOutputs() {
+ return outputs;
+ }
+
+ public void setOutputs(List outputs) {
+ this.outputs = outputs;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalToolBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalToolBlock.java
new file mode 100644
index 0000000..77b3586
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalToolBlock.java
@@ -0,0 +1,44 @@
+package ai.z.openapi.service.assistant.message.tools.retrieval;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+/**
+ * This class represents a block for invoking the retrieval tool.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("retrieval")
+public class RetrievalToolBlock {
+
+ /**
+ * An instance of the RetrievalTool class containing the retrieval outputs.
+ */
+ @JsonProperty("retrieval")
+ private RetrievalTool retrieval;
+
+ /**
+ * The type of tool being used, always set to "retrieval".
+ */
+ @JsonProperty("type")
+ private String type = "retrieval";
+
+ // Getters and Setters
+
+ public RetrievalTool getRetrieval() {
+ return retrieval;
+ }
+
+ public void setRetrieval(RetrievalTool retrieval) {
+ this.retrieval = retrieval;
+ }
+
+ public String getType() {
+ return type;
+ }
+
+ public void setType(String type) {
+ this.type = type;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalToolOutput.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalToolOutput.java
new file mode 100644
index 0000000..b70ce33
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/retrieval/RetrievalToolOutput.java
@@ -0,0 +1,41 @@
+package ai.z.openapi.service.assistant.message.tools.retrieval;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * This class represents the output of a retrieval tool.
+ */
+public class RetrievalToolOutput {
+
+ /**
+ * The text snippet retrieved from the knowledge base.
+ */
+ @JsonProperty("text")
+ private String text;
+
+ /**
+ * The name of the document from which the text snippet was retrieved, returned only
+ * in intelligent configuration.
+ */
+ @JsonProperty("document")
+ private String document;
+
+ // Getters and Setters
+
+ public String getText() {
+ return text;
+ }
+
+ public void setText(String text) {
+ this.text = text;
+ }
+
+ public String getDocument() {
+ return document;
+ }
+
+ public void setDocument(String document) {
+ this.document = document;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowser.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowser.java
new file mode 100644
index 0000000..51e39c4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowser.java
@@ -0,0 +1,41 @@
+package ai.z.openapi.service.assistant.message.tools.web_browser;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.List;
+
+/**
+ * This class represents the input and outputs of a web browser search.
+ */
+public class WebBrowser {
+
+ /**
+ * The input query for the web browser search.
+ */
+ @JsonProperty("input")
+ private String input;
+
+ /**
+ * A list of search results returned by the web browser.
+ */
+ @JsonProperty("outputs")
+ private List outputs;
+
+ // Getters and Setters
+
+ public String getInput() {
+ return input;
+ }
+
+ public void setInput(String input) {
+ this.input = input;
+ }
+
+ public List getOutputs() {
+ return outputs;
+ }
+
+ public void setOutputs(List outputs) {
+ this.outputs = outputs;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowserOutput.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowserOutput.java
new file mode 100644
index 0000000..1368b70
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowserOutput.java
@@ -0,0 +1,68 @@
+package ai.z.openapi.service.assistant.message.tools.web_browser;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * This class represents the output of a web browser search result.
+ */
+public class WebBrowserOutput {
+
+ /**
+ * The title of the search result.
+ */
+ @JsonProperty("title")
+ private String title;
+
+ /**
+ * The URL link to the search result's webpage.
+ */
+ @JsonProperty("link")
+ private String link;
+
+ /**
+ * The textual content extracted from the search result.
+ */
+ @JsonProperty("content")
+ private String content;
+
+ /**
+ * Any error message encountered during the search or retrieval process.
+ */
+ @JsonProperty("error_msg")
+ private String errorMsg;
+
+ // Getters and Setters
+
+ public String getTitle() {
+ return title;
+ }
+
+ public void setTitle(String title) {
+ this.title = title;
+ }
+
+ public String getLink() {
+ return link;
+ }
+
+ public void setLink(String link) {
+ this.link = link;
+ }
+
+ public String getContent() {
+ return content;
+ }
+
+ public void setContent(String content) {
+ this.content = content;
+ }
+
+ public String getErrorMsg() {
+ return errorMsg;
+ }
+
+ public void setErrorMsg(String errorMsg) {
+ this.errorMsg = errorMsg;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowserToolBlock.java b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowserToolBlock.java
new file mode 100644
index 0000000..e20d28f
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/message/tools/web_browser/WebBrowserToolBlock.java
@@ -0,0 +1,45 @@
+package ai.z.openapi.service.assistant.message.tools.web_browser;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+
+/**
+ * This class represents a block for invoking the web browser tool.
+ */
+@JsonIgnoreProperties(ignoreUnknown = true)
+@JsonTypeField("web_browser")
+public class WebBrowserToolBlock extends ToolsType {
+
+ /**
+ * An instance of the WebBrowser class containing the search input and outputs.
+ */
+ @JsonProperty("web_browser")
+ private WebBrowser webBrowser;
+
+ /**
+ * The type of tool being used, always set to "web_browser".
+ */
+ @JsonProperty("type")
+ private String type = "web_browser";
+
+ // Getters and Setters
+
+ public WebBrowser getWebBrowser() {
+ return webBrowser;
+ }
+
+ public void setWebBrowser(WebBrowser webBrowser) {
+ this.webBrowser = webBrowser;
+ }
+
+ public String getType() {
+ return type;
+ }
+
+ public void setType(String type) {
+ this.type = type;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupport.java b/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupport.java
new file mode 100644
index 0000000..5082e17
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupport.java
@@ -0,0 +1,70 @@
+package ai.z.openapi.service.assistant.query_support;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * This class represents the details of an assistant.
+ */
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class AssistantSupport {
+
+ /**
+ * The Assistant ID, used for assistant conversations.
+ */
+ @JsonProperty("assistant_id")
+ private String assistantId;
+
+ /**
+ * The creation time of the assistant.
+ */
+ @JsonProperty("created_at")
+ private String createdAt;
+
+ /**
+ * The last update time of the assistant.
+ */
+ @JsonProperty("updated_at")
+ private String updatedAt;
+
+ /**
+ * The name of the assistant.
+ */
+ @JsonProperty("name")
+ private String name;
+
+ /**
+ * The avatar of the assistant.
+ */
+ @JsonProperty("avatar")
+ private String avatar;
+
+ /**
+ * The description of the assistant.
+ */
+ @JsonProperty("description")
+ private String description;
+
+ /**
+ * The status of the assistant, currently only "publish".
+ */
+ @JsonProperty("status")
+ private String status;
+
+ /**
+ * The list of tools supported by the assistant.
+ */
+ @JsonProperty("tools")
+ private List tools;
+
+ /**
+ * The list of recommended prompts to start the assistant.
+ */
+ @JsonProperty("starter_prompts")
+ private List starterPrompts;
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupportResponse.java b/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupportResponse.java
new file mode 100644
index 0000000..1c09f4c
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupportResponse.java
@@ -0,0 +1,47 @@
+package ai.z.openapi.service.assistant.query_support;
+
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import lombok.Data;
+
+/**
+ * Response for assistant support query API calls. This class contains the response data
+ * for checking assistant support status.
+ */
+@Data
+public class AssistantSupportResponse implements ClientResponse {
+
+ /**
+ * Response status code.
+ */
+ private int code;
+
+ /**
+ * Response message.
+ */
+ private String msg;
+
+ /**
+ * Indicates if the request was successful.
+ */
+ private boolean success;
+
+ /**
+ * The assistant support status data.
+ */
+ private AssistantSupportStatus data;
+
+ /**
+ * Error information if the request failed.
+ */
+ private ChatError error;
+
+ public AssistantSupportResponse() {
+ }
+
+ public AssistantSupportResponse(int code, String msg) {
+ this.code = code;
+ this.msg = msg;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupportStatus.java b/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupportStatus.java
new file mode 100644
index 0000000..39aed27
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/query_support/AssistantSupportStatus.java
@@ -0,0 +1,34 @@
+package ai.z.openapi.service.assistant.query_support;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * This class represents the response containing a list of assistant supports.
+ */
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class AssistantSupportStatus {
+
+ /**
+ * The response code.
+ */
+ @JsonProperty("code")
+ private Integer code;
+
+ /**
+ * The response message.
+ */
+ @JsonProperty("msg")
+ private String msg;
+
+ /**
+ * The list of assistant supports.
+ */
+ @JsonProperty("data")
+ private List data;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/assistant/query_support/QuerySupportParams.java b/core/src/main/java/ai/z/openapi/service/assistant/query_support/QuerySupportParams.java
new file mode 100644
index 0000000..33fdd9c
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/assistant/query_support/QuerySupportParams.java
@@ -0,0 +1,31 @@
+package ai.z.openapi.service.assistant.query_support;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.List;
+
+/**
+ * Parameters for querying assistant support status. This class contains the parameters
+ * needed to query the support status of specific assistants.
+ */
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class QuerySupportParams extends CommonRequest implements ClientRequest {
+
+ /**
+ * List of assistant IDs to query support status for.
+ */
+ @JsonProperty("assistant_id_list")
+ private List assistantIdList;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioCustomizationApiResponse.java b/core/src/main/java/ai/z/openapi/service/audio/AudioCustomizationApiResponse.java
new file mode 100644
index 0000000..41eeca8
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioCustomizationApiResponse.java
@@ -0,0 +1,42 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import lombok.Data;
+
+import java.io.File;
+
+/**
+ * Response wrapper for audio customization API calls. This class contains the response
+ * data for audio customization requests, including the generated audio file and status
+ * information.
+ */
+@Data
+public class AudioCustomizationApiResponse implements ClientResponse {
+
+ /**
+ * Response status code.
+ */
+ private int code;
+
+ /**
+ * Response message.
+ */
+ private String msg;
+
+ /**
+ * Indicates whether the request was successful.
+ */
+ private boolean success;
+
+ /**
+ * The generated audio file.
+ */
+ private File data;
+
+ /**
+ * Error information if the request failed.
+ */
+ private ChatError error;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioCustomizationRequest.java b/core/src/main/java/ai/z/openapi/service/audio/AudioCustomizationRequest.java
new file mode 100644
index 0000000..03f51ca
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioCustomizationRequest.java
@@ -0,0 +1,63 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import ai.z.openapi.service.model.SensitiveWordCheckRequest;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.io.File;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Request parameters for audio customization API calls. This class contains all the
+ * necessary parameters for generating customized audio, including input text, model
+ * selection, voice cloning data, and response format options.
+ */
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AudioCustomizationRequest extends CommonRequest implements ClientRequest {
+
+ /**
+ * Text to generate audio from
+ */
+ private String input;
+
+ /**
+ * Model code to call
+ */
+ private String model;
+
+ /**
+ * Text description of the original audio to clone
+ */
+ @JsonProperty("voice_text")
+ private String voiceText;
+
+ /**
+ * Original audio file to clone
+ */
+ @JsonProperty("voice_data")
+ private File voiceData;
+
+ /**
+ * Audio response format
+ */
+ @JsonProperty("response_format")
+ private String responseFormat;
+
+ /**
+ * Sensitive word detection control
+ */
+ @JsonProperty("sensitive_word_check")
+ private SensitiveWordCheckRequest sensitiveWordCheck;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioService.java b/core/src/main/java/ai/z/openapi/service/audio/AudioService.java
new file mode 100644
index 0000000..86e6258
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioService.java
@@ -0,0 +1,31 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.service.model.ChatCompletionResponse;
+
+/**
+ * Audio service interface
+ */
+public interface AudioService {
+
+ /**
+ * Creates speech from text using text-to-speech.
+ * @param request the speech generation request
+ * @return AudioSpeechApiResponse containing the generated speech
+ */
+ AudioSpeechApiResponse createSpeech(AudioSpeechRequest request);
+
+ /**
+ * Creates customized speech with specific voice characteristics.
+ * @param request the speech customization request
+ * @return AudioCustomizationApiResponse containing the customized speech result
+ */
+ AudioCustomizationApiResponse createCustomSpeech(AudioCustomizationRequest request);
+
+ /**
+ * Creates audio transcriptions from audio files.
+ * @param request the transcription request
+ * @return ChatCompletionResponse containing the transcription result
+ */
+ ChatCompletionResponse createTranscription(AudioTranscriptionsRequest request);
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioServiceImpl.java b/core/src/main/java/ai/z/openapi/service/audio/AudioServiceImpl.java
new file mode 100644
index 0000000..4609d08
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioServiceImpl.java
@@ -0,0 +1,213 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.ZaiClient;
+import ai.z.openapi.api.audio.AudioApi;
+import ai.z.openapi.service.deserialize.MessageDeserializeFactory;
+import ai.z.openapi.service.model.ChatCompletionResponse;
+import ai.z.openapi.service.model.ModelData;
+import ai.z.openapi.utils.FlowableRequestSupplier;
+import ai.z.openapi.utils.RequestSupplier;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import io.reactivex.Single;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.MediaType;
+import okhttp3.MultipartBody;
+import okhttp3.RequestBody;
+import okhttp3.ResponseBody;
+import org.apache.tika.Tika;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
+
+/**
+ * Audio service implementation
+ */
+@Slf4j
+public class AudioServiceImpl implements AudioService {
+
+ protected static final ObjectMapper mapper = MessageDeserializeFactory.defaultObjectMapper();
+
+ private final ZaiClient zAiClient;
+
+ private final AudioApi audioApi;
+
+ public AudioServiceImpl(ZaiClient zAiClient) {
+ this.zAiClient = zAiClient;
+ this.audioApi = zAiClient.retrofit().create(AudioApi.class);
+ }
+
+ @Override
+ public AudioSpeechApiResponse createSpeech(AudioSpeechRequest request) {
+ RequestSupplier supplier = (params) -> {
+ try {
+ Single responseBody = audioApi.audioSpeech(params);
+ Path tempDirectory = Files.createTempFile("audio_speech" + UUID.randomUUID(), ".wav");
+ java.io.File file = tempDirectory.toFile();
+ writeResponseBodyToFile(responseBody.blockingGet(), file);
+ return Single.just(file);
+ }
+ catch (IOException e) {
+ throw new RuntimeException(e);
+ }
+ };
+ return this.zAiClient.executeRequest(request, supplier, AudioSpeechApiResponse.class);
+ }
+
+ @Override
+ public AudioCustomizationApiResponse createCustomSpeech(AudioCustomizationRequest request) {
+ RequestSupplier supplier = (params) -> {
+ try {
+ java.io.File voiceFile = params.getVoiceData();
+ RequestBody requestFile = RequestBody.create(MediaType.parse("multipart/form-data"), voiceFile);
+ MultipartBody.Part voiceData = MultipartBody.Part.createFormData("voice_data", voiceFile.getName(),
+ requestFile);
+
+ Map requestMap = new HashMap<>();
+ if (params.getInput() != null) {
+ requestMap.put("input", RequestBody.create(MediaType.parse("text/plain"), params.getInput()));
+ }
+ if (params.getModel() != null) {
+ requestMap.put("model", RequestBody.create(MediaType.parse("text/plain"), params.getModel()));
+ }
+ if (params.getVoiceText() != null) {
+ requestMap.put("voice_text",
+ RequestBody.create(MediaType.parse("text/plain"), params.getVoiceText()));
+ }
+ if (params.getResponseFormat() != null) {
+ requestMap.put("response_format",
+ RequestBody.create(MediaType.parse("text/plain"), params.getResponseFormat()));
+ }
+ if (params.getSensitiveWordCheck() != null) {
+ try {
+ String sensitiveWordCheckJson = mapper.writeValueAsString(params.getSensitiveWordCheck());
+ requestMap.put("sensitive_word_check",
+ RequestBody.create(MediaType.parse("application/json"), sensitiveWordCheckJson));
+ }
+ catch (Exception e) {
+ log.error("Error serializing sensitive_word_check: {}", e.getMessage(), e);
+ }
+ }
+ if (params.getRequestId() != null) {
+ requestMap.put("request_id",
+ RequestBody.create(MediaType.parse("text/plain"), params.getRequestId()));
+ }
+ if (params.getUserId() != null) {
+ requestMap.put("user_id", RequestBody.create(MediaType.parse("text/plain"), params.getUserId()));
+ }
+
+ Single responseBody = audioApi.audioCustomization(requestMap, voiceData);
+ Path tempDirectory = Files.createTempFile("audio_customization" + UUID.randomUUID(), ".wav");
+ java.io.File file = tempDirectory.toFile();
+ writeResponseBodyToFile(responseBody.blockingGet(), file);
+ return Single.just(file);
+ }
+ catch (IOException e) {
+ log.error("Error create custom speak: {}", e.getMessage(), e);
+ throw new RuntimeException(e);
+ }
+ };
+ return this.zAiClient.executeRequest(request, supplier, AudioCustomizationApiResponse.class);
+ }
+
+ @Override
+ public ChatCompletionResponse createTranscription(AudioTranscriptionsRequest request) {
+ if (request.getStream()) {
+ return createTranscriptionStream(request);
+ }
+ else {
+ return createTranscriptionSync(request);
+ }
+ }
+
+ private ChatCompletionResponse createTranscriptionStream(AudioTranscriptionsRequest request) {
+ FlowableRequestSupplier> supplier = params -> {
+ try {
+ java.io.File file = params.getFile();
+ Tika tika = new Tika();
+ String contentType = tika.detect(file);
+ RequestBody requestFile = RequestBody.create(MediaType.parse(contentType), file);
+ MultipartBody.Part fileData = MultipartBody.Part.createFormData("file", file.getName(), requestFile);
+
+ Map requestMap = new HashMap<>();
+ if (params.getModel() != null) {
+ requestMap.put("model", RequestBody.create(MediaType.parse("text/plain"), params.getModel()));
+ }
+ if (params.getStream() != null) {
+ requestMap.put("stream",
+ RequestBody.create(MediaType.parse("text/plain"), params.getStream().toString()));
+ }
+ if (params.getRequestId() != null) {
+ requestMap.put("request_id",
+ RequestBody.create(MediaType.parse("text/plain"), params.getRequestId()));
+ }
+ if (params.getUserId() != null) {
+ requestMap.put("user_id", RequestBody.create(MediaType.parse("text/plain"), params.getUserId()));
+ }
+
+ return audioApi.audioTranscriptionsStream(requestMap, fileData);
+ }
+ catch (IOException e) {
+ log.error("Error create transcription: {}", e.getMessage(), e);
+ throw new RuntimeException(e);
+ }
+ };
+ return this.zAiClient.streamRequest(request, supplier, ChatCompletionResponse.class, ModelData.class);
+ }
+
+ private ChatCompletionResponse createTranscriptionSync(AudioTranscriptionsRequest request) {
+ RequestSupplier supplier = (params) -> {
+ try {
+ java.io.File file = params.getFile();
+ Tika tika = new Tika();
+ String contentType = tika.detect(file);
+ RequestBody requestFile = RequestBody.create(MediaType.parse(contentType), file);
+ MultipartBody.Part fileData = MultipartBody.Part.createFormData("file", file.getName(), requestFile);
+
+ Map requestMap = new HashMap<>();
+ if (params.getModel() != null) {
+ requestMap.put("model", RequestBody.create(MediaType.parse("text/plain"), params.getModel()));
+ }
+ if (params.getStream() != null) {
+ requestMap.put("stream",
+ RequestBody.create(MediaType.parse("text/plain"), params.getStream().toString()));
+ }
+ if (params.getRequestId() != null) {
+ requestMap.put("request_id",
+ RequestBody.create(MediaType.parse("text/plain"), params.getRequestId()));
+ }
+ if (params.getUserId() != null) {
+ requestMap.put("user_id", RequestBody.create(MediaType.parse("text/plain"), params.getUserId()));
+ }
+
+ return audioApi.audioTranscriptions(requestMap, fileData);
+ }
+ catch (IOException e) {
+ log.error("Error create transcription: {}", e.getMessage(), e);
+ throw new RuntimeException(e);
+ }
+ };
+ return this.zAiClient.executeRequest(request, supplier, ChatCompletionResponse.class);
+ }
+
+ private void writeResponseBodyToFile(ResponseBody body, java.io.File file) {
+ try (InputStream inputStream = body.byteStream();
+ OutputStream outputStream = Files.newOutputStream(file.toPath())) {
+ byte[] buffer = new byte[4096];
+ int bytesRead;
+ while ((bytesRead = inputStream.read(buffer)) != -1) {
+ outputStream.write(buffer, 0, bytesRead);
+ }
+ outputStream.flush();
+ }
+ catch (IOException e) {
+ log.error("writeResponseBodyToFile error,msg:{}", e.getMessage(), e);
+ }
+ }
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioSpeechApiResponse.java b/core/src/main/java/ai/z/openapi/service/audio/AudioSpeechApiResponse.java
new file mode 100644
index 0000000..f7cffbc
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioSpeechApiResponse.java
@@ -0,0 +1,22 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import lombok.Data;
+
+import java.io.File;
+
+@Data
+public class AudioSpeechApiResponse implements ClientResponse {
+
+ private int code;
+
+ private String msg;
+
+ private boolean success;
+
+ private File data;
+
+ private ChatError error;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioSpeechRequest.java b/core/src/main/java/ai/z/openapi/service/audio/AudioSpeechRequest.java
new file mode 100644
index 0000000..db7cede
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioSpeechRequest.java
@@ -0,0 +1,50 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import ai.z.openapi.service.model.SensitiveWordCheckRequest;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AudioSpeechRequest extends CommonRequest implements ClientRequest {
+
+ /**
+ * Model code to call
+ */
+ private String model;
+
+ /**
+ * Text to generate speech from
+ */
+ private String input;
+
+ /**
+ * Voice tone for speech generation
+ */
+ private String voice;
+
+ /**
+ * Format of the generated speech file
+ */
+ @JsonProperty("response_format")
+ private String responseFormat;
+
+ /**
+ * Sensitive word detection control
+ */
+ @JsonProperty("sensitive_word_check")
+ private SensitiveWordCheckRequest sensitiveWordCheck;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/audio/AudioTranscriptionsRequest.java b/core/src/main/java/ai/z/openapi/service/audio/AudioTranscriptionsRequest.java
new file mode 100644
index 0000000..f5ec9b2
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/audio/AudioTranscriptionsRequest.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.audio;
+
+import ai.z.openapi.core.model.ClientRequest;
+import ai.z.openapi.service.CommonRequest;
+import ai.z.openapi.service.model.SensitiveWordCheckRequest;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.io.File;
+import java.util.HashMap;
+import java.util.Map;
+
+@EqualsAndHashCode(callSuper = true)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class AudioTranscriptionsRequest extends CommonRequest implements ClientRequest {
+
+ /**
+ * Model code to call
+ */
+ private String model;
+
+ /**
+ * Synchronous call: false, SSE call: true
+ */
+ private Boolean stream;
+
+ private File file;
+
+ /**
+ * Sampling temperature, controls output randomness, must be positive Range:
+ * (0.0,1.0], cannot equal 0, default value is 0.95 Higher values make output more
+ * random and creative; lower values make output more stable or deterministic It's
+ * recommended to adjust either top_p or temperature parameter based on your use case,
+ * but not both simultaneously
+ */
+ private Float temperature;
+
+ /**
+ * Sensitive word detection control
+ */
+ @JsonProperty("sensitive_word_check")
+ private SensitiveWordCheckRequest sensitiveWordCheck;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/batches/Batch.java b/core/src/main/java/ai/z/openapi/service/batches/Batch.java
new file mode 100644
index 0000000..937dcd9
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/Batch.java
@@ -0,0 +1,113 @@
+package ai.z.openapi.service.batches;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+public class Batch {
+
+ @JsonProperty("id")
+ private String id;
+
+ @JsonProperty("completion_window")
+ private String completionWindow;
+
+ @JsonProperty("created_at")
+ private long createdAt;
+
+ @JsonProperty("endpoint")
+ private String endpoint;
+
+ @JsonProperty("input_file_id")
+ private String inputFileId;
+
+ @JsonProperty("object")
+ private String object;
+
+ @JsonProperty("status")
+ private String status;
+
+ @JsonProperty("cancelled_at")
+ private Long cancelledAt;
+
+ @JsonProperty("cancelling_at")
+ private Long cancellingAt;
+
+ @JsonProperty("completed_at")
+ private Long completedAt;
+
+ @JsonProperty("error_file_id")
+ private String errorFileId;
+
+ @JsonProperty("errors")
+ private Errors errors;
+
+ @JsonProperty("expired_at")
+ private Long expiredAt;
+
+ @JsonProperty("expires_at")
+ private Long expiresAt;
+
+ @JsonProperty("failed_at")
+ private Long failedAt;
+
+ @JsonProperty("finalizing_at")
+ private Long finalizingAt;
+
+ @JsonProperty("in_progress_at")
+ private Long inProgressAt;
+
+ @JsonProperty("metadata")
+ private Object metadata;
+
+ @JsonProperty("output_file_id")
+ private String outputFileId;
+
+ @JsonProperty("request_counts")
+ private BatchRequestCounts requestCounts;
+
+}
+
+@Data
+class Errors {
+
+ @JsonProperty("data")
+ private List data;
+
+ @JsonProperty("object")
+ private String object;
+
+}
+
+@Data
+class BatchRequestCounts {
+
+ @JsonProperty("completed")
+ private int completed;
+
+ @JsonProperty("failed")
+ private int failed;
+
+ @JsonProperty("total")
+ private int total;
+
+}
+
+@Data
+class BatchError {
+
+ @JsonProperty("code")
+ private String code;
+
+ @JsonProperty("line")
+ private Long line;
+
+ @JsonProperty("message")
+ private String message;
+
+ @JsonProperty("param")
+ private String param;
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/batches/BatchCreateParams.java b/core/src/main/java/ai/z/openapi/service/batches/BatchCreateParams.java
new file mode 100644
index 0000000..3dc62d2
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/BatchCreateParams.java
@@ -0,0 +1,56 @@
+package ai.z.openapi.service.batches;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import ai.z.openapi.core.model.ClientRequest;
+import lombok.*;
+import lombok.experimental.SuperBuilder;
+
+import java.util.Map;
+
+/**
+ * Parameters for creating batch processing requests. This class contains all the
+ * necessary parameters to create a batch job, including completion window, endpoint,
+ * input file, and metadata.
+ */
+@EqualsAndHashCode(callSuper = false)
+@SuperBuilder
+@Data
+public class BatchCreateParams implements ClientRequest {
+
+ /**
+ * The time frame within which the batch should be processed.
+ */
+ @JsonProperty("completion_window")
+ private String completionWindow;
+
+ /**
+ * The API endpoint to be used for batch processing.
+ */
+ @JsonProperty("endpoint")
+ private String endpoint;
+
+ /**
+ * The ID of the uploaded file containing batch requests. Must be the ID of the
+ * uploaded file.
+ */
+ @JsonProperty("input_file_id")
+ private String inputFileId;
+
+ /**
+ * Optional custom metadata for the batch job.
+ */
+ @JsonProperty("metadata")
+ private Map metadata;
+
+ public BatchCreateParams() {
+ }
+
+ public BatchCreateParams(String completionWindow, String endpoint, String inputFileId,
+ Map metadata) {
+ this.completionWindow = completionWindow;
+ this.endpoint = endpoint;
+ this.inputFileId = inputFileId;
+ this.metadata = metadata;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/batches/BatchPage.java b/core/src/main/java/ai/z/openapi/service/batches/BatchPage.java
new file mode 100644
index 0000000..6247438
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/BatchPage.java
@@ -0,0 +1,16 @@
+package ai.z.openapi.service.batches;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class BatchPage {
+
+ private String object;
+
+ private List data;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/batches/BatchRequest.java b/core/src/main/java/ai/z/openapi/service/batches/BatchRequest.java
new file mode 100644
index 0000000..edc9d1d
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/BatchRequest.java
@@ -0,0 +1,19 @@
+package ai.z.openapi.service.batches;
+
+import ai.z.openapi.core.model.ClientRequest;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+@EqualsAndHashCode(callSuper = false)
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+public class BatchRequest implements ClientRequest {
+
+ private String batchId;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/batches/BatchResponse.java b/core/src/main/java/ai/z/openapi/service/batches/BatchResponse.java
new file mode 100644
index 0000000..5e87fa5
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/BatchResponse.java
@@ -0,0 +1,20 @@
+package ai.z.openapi.service.batches;
+
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import lombok.Data;
+
+@Data
+public class BatchResponse implements ClientResponse {
+
+ private int code;
+
+ private String msg;
+
+ private boolean success;
+
+ private Batch data;
+
+ private ChatError error;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/batches/BatchService.java b/core/src/main/java/ai/z/openapi/service/batches/BatchService.java
new file mode 100644
index 0000000..69be046
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/BatchService.java
@@ -0,0 +1,52 @@
+package ai.z.openapi.service.batches;
+
+import ai.z.openapi.service.file.QueryBatchRequest;
+
+/**
+ * Batch service interface
+ */
+public interface BatchService {
+
+ /**
+ * Creates a batch request.
+ * @param batchCreateParams the batch creation parameters
+ * @return BatchResponse containing the created batch details
+ */
+ BatchResponse createBatch(BatchCreateParams batchCreateParams);
+
+ /**
+ * Retrieves a specific batch.
+ * @param request the batch request containing the batch ID
+ * @return BatchResponse containing the batch details
+ */
+ BatchResponse retrieveBatch(BatchRequest request);
+
+ /**
+ * Retrieves a specific batch by ID.
+ * @param batchId the ID of the batch to retrieve
+ * @return BatchResponse containing the batch details
+ */
+ BatchResponse retrieveBatch(String batchId);
+
+ /**
+ * Lists all batches.
+ * @param queryBatchRequest the query parameters for listing batches
+ * @return QueryBatchResponse containing the list of batches
+ */
+ QueryBatchResponse listBatches(QueryBatchRequest queryBatchRequest);
+
+ /**
+ * Cancels a batch request.
+ * @param request the batch request containing the batch ID
+ * @return BatchResponse containing the cancellation result
+ */
+ BatchResponse cancelBatch(BatchRequest request);
+
+ /**
+ * Cancels a batch request by ID.
+ * @param batchId the ID of the batch to cancel
+ * @return BatchResponse containing the cancellation result
+ */
+ BatchResponse cancelBatch(String batchId);
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/batches/BatchServiceImpl.java b/core/src/main/java/ai/z/openapi/service/batches/BatchServiceImpl.java
new file mode 100644
index 0000000..c781d7d
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/BatchServiceImpl.java
@@ -0,0 +1,59 @@
+package ai.z.openapi.service.batches;
+
+import ai.z.openapi.ZaiClient;
+import ai.z.openapi.api.batches.BatchesApi;
+import ai.z.openapi.service.file.QueryBatchRequest;
+import ai.z.openapi.utils.RequestSupplier;
+
+/**
+ * Batch service implementation
+ */
+public class BatchServiceImpl implements BatchService {
+
+ private final ZaiClient zAiClient;
+
+ private final BatchesApi batchesApi;
+
+ public BatchServiceImpl(ZaiClient zAiClient) {
+ this.zAiClient = zAiClient;
+ this.batchesApi = zAiClient.retrofit().create(BatchesApi.class);
+ }
+
+ @Override
+ public BatchResponse createBatch(BatchCreateParams batchCreateParams) {
+ RequestSupplier supplier = batchesApi::batchesCreate;
+ return this.zAiClient.executeRequest(batchCreateParams, supplier, BatchResponse.class);
+ }
+
+ @Override
+ public BatchResponse retrieveBatch(BatchRequest request) {
+ RequestSupplier supplier = (params) -> batchesApi.batchesRetrieve(params.getBatchId());
+ return this.zAiClient.executeRequest(request, supplier, BatchResponse.class);
+ }
+
+ @Override
+ public BatchResponse retrieveBatch(String batchId) {
+ BatchRequest request = BatchRequest.builder().batchId(batchId).build();
+ return retrieveBatch(request);
+ }
+
+ @Override
+ public QueryBatchResponse listBatches(QueryBatchRequest queryBatchRequest) {
+ RequestSupplier supplier = (params) -> batchesApi.batchesList(params.getAfter(),
+ params.getLimit());
+ return this.zAiClient.executeRequest(queryBatchRequest, supplier, QueryBatchResponse.class);
+ }
+
+ @Override
+ public BatchResponse cancelBatch(BatchRequest request) {
+ RequestSupplier supplier = (params) -> batchesApi.batchesCancel(params.getBatchId());
+ return this.zAiClient.executeRequest(request, supplier, BatchResponse.class);
+ }
+
+ @Override
+ public BatchResponse cancelBatch(String batchId) {
+ BatchRequest request = BatchRequest.builder().batchId(batchId).build();
+ return cancelBatch(request);
+ }
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/batches/QueryBatchResponse.java b/core/src/main/java/ai/z/openapi/service/batches/QueryBatchResponse.java
new file mode 100644
index 0000000..e24ac54
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/batches/QueryBatchResponse.java
@@ -0,0 +1,20 @@
+package ai.z.openapi.service.batches;
+
+import ai.z.openapi.core.model.ClientResponse;
+import ai.z.openapi.service.model.ChatError;
+import lombok.Data;
+
+@Data
+public class QueryBatchResponse implements ClientResponse {
+
+ private int code;
+
+ private String msg;
+
+ private boolean success;
+
+ private BatchPage data;
+
+ private ChatError error;
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/chat/ChatService.java b/core/src/main/java/ai/z/openapi/service/chat/ChatService.java
new file mode 100644
index 0000000..f382290
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/chat/ChatService.java
@@ -0,0 +1,35 @@
+package ai.z.openapi.service.chat;
+
+import ai.z.openapi.service.model.ChatCompletionCreateParams;
+import ai.z.openapi.service.model.ChatCompletionResponse;
+import ai.z.openapi.service.model.AsyncResultRetrieveParams;
+import ai.z.openapi.service.model.QueryModelResultResponse;
+
+/**
+ * Chat completion service interface
+ */
+public interface ChatService {
+
+ /**
+ * Creates a chat completion, either streaming or non-streaming based on the request
+ * configuration.
+ * @param request the chat completion request
+ * @return ChatCompletionResponse containing the completion result
+ */
+ ChatCompletionResponse createChatCompletion(ChatCompletionCreateParams request);
+
+ /**
+ * Creates an asynchronous chat completion.
+ * @param request the chat completion request
+ * @return ChatCompletionResponse containing the async completion result
+ */
+ ChatCompletionResponse asyncChatCompletion(ChatCompletionCreateParams request);
+
+ /**
+ * Retrieves the result of an asynchronous model operation.
+ * @param request the query request for the async result
+ * @return QueryModelResultResponse containing the async operation result
+ */
+ QueryModelResultResponse retrieveAsyncResult(AsyncResultRetrieveParams request);
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/chat/ChatServiceImpl.java b/core/src/main/java/ai/z/openapi/service/chat/ChatServiceImpl.java
new file mode 100644
index 0000000..cc58763
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/chat/ChatServiceImpl.java
@@ -0,0 +1,81 @@
+package ai.z.openapi.service.chat;
+
+import ai.z.openapi.ZaiClient;
+import ai.z.openapi.api.chat.ChatApi;
+import ai.z.openapi.service.model.ChatCompletionCreateParams;
+import ai.z.openapi.service.model.ChatCompletionResponse;
+import ai.z.openapi.service.model.AsyncResultRetrieveParams;
+import ai.z.openapi.service.model.QueryModelResultResponse;
+import ai.z.openapi.service.model.ModelData;
+import ai.z.openapi.utils.FlowableRequestSupplier;
+import ai.z.openapi.utils.RequestSupplier;
+import ai.z.openapi.utils.StringUtils;
+import okhttp3.ResponseBody;
+
+/**
+ * Chat completion service implementation
+ */
+public class ChatServiceImpl implements ChatService {
+
+ private final ZaiClient zAiClient;
+
+ private final ChatApi chatApi;
+
+ public ChatServiceImpl(ZaiClient zAiClient) {
+ this.zAiClient = zAiClient;
+ this.chatApi = this.zAiClient.retrofit().create(ChatApi.class);
+ }
+
+ @Override
+ public ChatCompletionResponse createChatCompletion(ChatCompletionCreateParams request) {
+ String paramMsg = validateParams(request);
+ if (StringUtils.isNotEmpty(paramMsg)) {
+ return new ChatCompletionResponse(-100, String.format("invalid param: %s", paramMsg));
+ }
+ if (request.getStream()) {
+ return streamChatCompletion(request);
+ }
+ else {
+ return syncChatCompletion(request);
+ }
+ }
+
+ @Override
+ public ChatCompletionResponse asyncChatCompletion(ChatCompletionCreateParams request) {
+ RequestSupplier supplier = chatApi::createChatCompletionAsync;
+ return this.zAiClient.executeRequest(request, supplier, ChatCompletionResponse.class);
+ }
+
+ @Override
+ public QueryModelResultResponse retrieveAsyncResult(AsyncResultRetrieveParams request) {
+ RequestSupplier supplier = (params) -> chatApi
+ .queryAsyncResult(params.getTaskId());
+ // Handle response
+ return this.zAiClient.executeRequest(request, supplier, QueryModelResultResponse.class);
+ }
+
+ private ChatCompletionResponse streamChatCompletion(ChatCompletionCreateParams request) {
+ FlowableRequestSupplier> supplier = chatApi::createChatCompletionStream;
+ return this.zAiClient.streamRequest(request, supplier, ChatCompletionResponse.class, ModelData.class);
+ }
+
+ private ChatCompletionResponse syncChatCompletion(ChatCompletionCreateParams request) {
+ RequestSupplier supplier = chatApi::createChatCompletion;
+ // Handle response
+ return this.zAiClient.executeRequest(request, supplier, ChatCompletionResponse.class);
+ }
+
+ private String validateParams(ChatCompletionCreateParams request) {
+ if (request == null) {
+ return "request can not be null";
+ }
+ if (request.getMessages() == null || request.getMessages().isEmpty()) {
+ return "message can not be empty";
+ }
+ if (request.getModel() == null) {
+ return "model can not be empty";
+ }
+ return null;
+ }
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/BaseNodeDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/BaseNodeDeserializer.java
new file mode 100644
index 0000000..fd6a14e
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/BaseNodeDeserializer.java
@@ -0,0 +1,461 @@
+package ai.z.openapi.service.deserialize;
+
+import java.io.IOException;
+
+import com.fasterxml.jackson.databind.*;
+import com.fasterxml.jackson.databind.jsontype.TypeDeserializer;
+import com.fasterxml.jackson.databind.util.RawValue;
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.core.JsonToken;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.JsonNodeFactory;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+/**
+ * Base class for all actual {@link JsonNode} deserializer implementations
+ */
+public abstract class BaseNodeDeserializer extends StdDeserializer {
+
+ protected final Boolean _supportsUpdates;
+
+ public BaseNodeDeserializer(Class vc, Boolean supportsUpdates) {
+ super(vc);
+ _supportsUpdates = supportsUpdates;
+ }
+
+ @Override
+ public Object deserializeWithType(JsonParser p, DeserializationContext ctxt, TypeDeserializer typeDeserializer)
+ throws IOException {
+ // Output can be as JSON Object, Array or scalar: no way to know a priori:
+ return typeDeserializer.deserializeTypedFromAny(p, ctxt);
+ }
+
+ /*
+ * 07-Nov-2014, tatu: When investigating [databind#604], realized that it makes sense
+ * to also mark this is cachable, since lookup not exactly free, and since it's not
+ * uncommon to "read anything"
+ */
+ @Override
+ public boolean isCachable() {
+ return true;
+ }
+
+ @Override // since 2.9
+ public Boolean supportsUpdate(DeserializationConfig config) {
+ return _supportsUpdates;
+ }
+
+ /*
+ * /********************************************************** /* Overridable methods
+ * /**********************************************************
+ */
+
+ /**
+ * Method called when there is a duplicate value for a field. By default we don't
+ * care, and the last value is used. Can be overridden to provide alternate handling,
+ * such as throwing an exception, or choosing different strategy for combining values
+ * or choosing which one to keep.
+ * @param fieldName Name of the field for which duplicate value was found
+ * @param objectNode Object node that contains values
+ * @param oldValue Value that existed for the object node before newValue was added
+ * @param newValue Newly added value just added to the object node
+ */
+ protected void _handleDuplicateField(JsonParser p, DeserializationContext ctxt, JsonNodeFactory nodeFactory,
+ String fieldName, ObjectNode objectNode, JsonNode oldValue, JsonNode newValue)
+ throws JsonProcessingException {
+ // [databind#237]: Report an error if asked to do so:
+ if (ctxt.isEnabled(DeserializationFeature.FAIL_ON_READING_DUP_TREE_KEY)) {
+ // 11-Sep-2019, tatu: Can not pass "property name" because we may be
+ // missing enclosing JSON content context...
+ // ctxt.reportPropertyInputMismatch(JsonNode.class, fieldName,
+ ctxt.reportInputMismatch(JsonNode.class,
+ "Duplicate field '%s' for `ObjectNode`: not allowed when `DeserializationFeature.FAIL_ON_READING_DUP_TREE_KEY` enabled",
+ fieldName);
+ }
+ }
+
+ /*
+ * /********************************************************** /* Helper methods
+ * /**********************************************************
+ */
+
+ /**
+ * Method called to deserialize Object node instance when there is no existing node to
+ * modify.
+ */
+ protected final ObjectNode deserializeObject(JsonParser p, DeserializationContext ctxt,
+ final JsonNodeFactory nodeFactory) throws IOException {
+ final ObjectNode node = nodeFactory.objectNode();
+ String key = p.nextFieldName();
+ for (; key != null; key = p.nextFieldName()) {
+ JsonNode value;
+ JsonToken t = p.nextToken();
+ if (t == null) { // can this ever occur?
+ t = JsonToken.NOT_AVAILABLE; // can this ever occur?
+ }
+ switch (t.id()) {
+ case JsonTokenId.ID_START_OBJECT:
+ value = deserializeObject(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_START_ARRAY:
+ value = deserializeArray(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_EMBEDDED_OBJECT:
+ value = _fromEmbedded(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_STRING:
+ value = nodeFactory.textNode(p.getText());
+ break;
+ case JsonTokenId.ID_NUMBER_INT:
+ value = _fromInt(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_TRUE:
+ value = nodeFactory.booleanNode(true);
+ break;
+ case JsonTokenId.ID_FALSE:
+ value = nodeFactory.booleanNode(false);
+ break;
+ case JsonTokenId.ID_NULL:
+ value = nodeFactory.nullNode();
+ break;
+ default:
+ value = deserializeAny(p, ctxt, nodeFactory);
+ }
+ JsonNode old = node.replace(key, value);
+ if (old != null) {
+ _handleDuplicateField(p, ctxt, nodeFactory, key, node, old, value);
+ }
+ }
+ return node;
+ }
+
+ /**
+ * Alternate deserialization method used when parser already points to first
+ * FIELD_NAME and not START_OBJECT.
+ *
+ * @since 2.9
+ */
+ protected final ObjectNode deserializeObjectAtName(JsonParser p, DeserializationContext ctxt,
+ final JsonNodeFactory nodeFactory) throws IOException {
+ final ObjectNode node = nodeFactory.objectNode();
+ String key = p.getCurrentName();
+ for (; key != null; key = p.nextFieldName()) {
+ JsonNode value;
+ JsonToken t = p.nextToken();
+ if (t == null) { // can this ever occur?
+ t = JsonToken.NOT_AVAILABLE; // can this ever occur?
+ }
+ switch (t.id()) {
+ case JsonTokenId.ID_START_OBJECT:
+ value = deserializeObject(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_START_ARRAY:
+ value = deserializeArray(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_EMBEDDED_OBJECT:
+ value = _fromEmbedded(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_STRING:
+ value = nodeFactory.textNode(p.getText());
+ break;
+ case JsonTokenId.ID_NUMBER_INT:
+ value = _fromInt(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_TRUE:
+ value = nodeFactory.booleanNode(true);
+ break;
+ case JsonTokenId.ID_FALSE:
+ value = nodeFactory.booleanNode(false);
+ break;
+ case JsonTokenId.ID_NULL:
+ value = nodeFactory.nullNode();
+ break;
+ default:
+ value = deserializeAny(p, ctxt, nodeFactory);
+ }
+ JsonNode old = node.replace(key, value);
+ if (old != null) {
+ _handleDuplicateField(p, ctxt, nodeFactory, key, node, old, value);
+ }
+ }
+ return node;
+ }
+
+ /**
+ * Alternate deserialization method that is to update existing {@link ObjectNode} if
+ * possible.
+ *
+ * @since 2.9
+ */
+ protected final JsonNode updateObject(JsonParser p, DeserializationContext ctxt, final ObjectNode node)
+ throws IOException {
+ String key;
+ if (p.isExpectedStartObjectToken()) {
+ key = p.nextFieldName();
+ }
+ else {
+ if (!p.hasToken(JsonToken.FIELD_NAME)) {
+ return deserialize(p, ctxt);
+ }
+ key = p.getCurrentName();
+ }
+ for (; key != null; key = p.nextFieldName()) {
+ // If not, fall through to regular handling
+ JsonToken t = p.nextToken();
+
+ // First: see if we can merge things:
+ JsonNode old = node.get(key);
+ if (old != null) {
+ if (old instanceof ObjectNode) {
+ JsonNode newValue = updateObject(p, ctxt, (ObjectNode) old);
+ if (newValue != old) {
+ node.set(key, newValue);
+ }
+ continue;
+ }
+ if (old instanceof ArrayNode) {
+ JsonNode newValue = updateArray(p, ctxt, (ArrayNode) old);
+ if (newValue != old) {
+ node.set(key, newValue);
+ }
+ continue;
+ }
+ }
+ if (t == null) { // can this ever occur?
+ t = JsonToken.NOT_AVAILABLE;
+ }
+ JsonNode value;
+ JsonNodeFactory nodeFactory = ctxt.getNodeFactory();
+ switch (t.id()) {
+ case JsonTokenId.ID_START_OBJECT:
+ value = deserializeObject(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_START_ARRAY:
+ value = deserializeArray(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_EMBEDDED_OBJECT:
+ value = _fromEmbedded(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_STRING:
+ value = nodeFactory.textNode(p.getText());
+ break;
+ case JsonTokenId.ID_NUMBER_INT:
+ value = _fromInt(p, ctxt, nodeFactory);
+ break;
+ case JsonTokenId.ID_TRUE:
+ value = nodeFactory.booleanNode(true);
+ break;
+ case JsonTokenId.ID_FALSE:
+ value = nodeFactory.booleanNode(false);
+ break;
+ case JsonTokenId.ID_NULL:
+ value = nodeFactory.nullNode();
+ break;
+ default:
+ value = deserializeAny(p, ctxt, nodeFactory);
+ }
+ if (old != null) {
+ _handleDuplicateField(p, ctxt, nodeFactory, key, node, old, value);
+ }
+ node.set(key, value);
+ }
+ return node;
+ }
+
+ protected final ArrayNode deserializeArray(JsonParser p, DeserializationContext ctxt,
+ final JsonNodeFactory nodeFactory) throws IOException {
+ ArrayNode node = nodeFactory.arrayNode();
+ while (true) {
+ JsonToken t = p.nextToken();
+ switch (t.id()) {
+ case JsonTokenId.ID_START_OBJECT:
+ node.add(deserializeObject(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_START_ARRAY:
+ node.add(deserializeArray(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_END_ARRAY:
+ return node;
+ case JsonTokenId.ID_EMBEDDED_OBJECT:
+ node.add(_fromEmbedded(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_STRING:
+ node.add(nodeFactory.textNode(p.getText()));
+ break;
+ case JsonTokenId.ID_NUMBER_INT:
+ node.add(_fromInt(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_TRUE:
+ node.add(nodeFactory.booleanNode(true));
+ break;
+ case JsonTokenId.ID_FALSE:
+ node.add(nodeFactory.booleanNode(false));
+ break;
+ case JsonTokenId.ID_NULL:
+ node.add(nodeFactory.nullNode());
+ break;
+ default:
+ node.add(deserializeAny(p, ctxt, nodeFactory));
+ break;
+ }
+ }
+ }
+
+ /**
+ * Alternate deserialization method that is to update existing {@link ObjectNode} if
+ * possible.
+ *
+ * @since 2.9
+ */
+ protected final JsonNode updateArray(JsonParser p, DeserializationContext ctxt, final ArrayNode node)
+ throws IOException {
+ final JsonNodeFactory nodeFactory = ctxt.getNodeFactory();
+ while (true) {
+ JsonToken t = p.nextToken();
+ switch (t.id()) {
+ case JsonTokenId.ID_START_OBJECT:
+ node.add(deserializeObject(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_START_ARRAY:
+ node.add(deserializeArray(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_END_ARRAY:
+ return node;
+ case JsonTokenId.ID_EMBEDDED_OBJECT:
+ node.add(_fromEmbedded(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_STRING:
+ node.add(nodeFactory.textNode(p.getText()));
+ break;
+ case JsonTokenId.ID_NUMBER_INT:
+ node.add(_fromInt(p, ctxt, nodeFactory));
+ break;
+ case JsonTokenId.ID_TRUE:
+ node.add(nodeFactory.booleanNode(true));
+ break;
+ case JsonTokenId.ID_FALSE:
+ node.add(nodeFactory.booleanNode(false));
+ break;
+ case JsonTokenId.ID_NULL:
+ node.add(nodeFactory.nullNode());
+ break;
+ default:
+ node.add(deserializeAny(p, ctxt, nodeFactory));
+ break;
+ }
+ }
+ }
+
+ protected final JsonNode deserializeAny(JsonParser p, DeserializationContext ctxt,
+ final JsonNodeFactory nodeFactory) throws IOException {
+ switch (p.currentTokenId()) {
+ case JsonTokenId.ID_END_OBJECT: // for empty JSON Objects we may point to
+ // this?
+ return nodeFactory.objectNode();
+ case JsonTokenId.ID_FIELD_NAME:
+ return deserializeObjectAtName(p, ctxt, nodeFactory);
+ case JsonTokenId.ID_EMBEDDED_OBJECT:
+ return _fromEmbedded(p, ctxt, nodeFactory);
+ case JsonTokenId.ID_STRING:
+ return nodeFactory.textNode(p.getText());
+ case JsonTokenId.ID_NUMBER_INT:
+ return _fromInt(p, ctxt, nodeFactory);
+ case JsonTokenId.ID_NUMBER_FLOAT:
+ return _fromFloat(p, ctxt, nodeFactory);
+ case JsonTokenId.ID_TRUE:
+ return nodeFactory.booleanNode(true);
+ case JsonTokenId.ID_FALSE:
+ return nodeFactory.booleanNode(false);
+ case JsonTokenId.ID_NULL:
+ return nodeFactory.nullNode();
+
+ /*
+ * Caller checks for these, should not get here ever case
+ * JsonTokenId.ID_START_OBJECT: return deserializeObject(p, ctxt,
+ * nodeFactory); case JsonTokenId.ID_START_ARRAY: return deserializeArray(p,
+ * ctxt, nodeFactory);
+ */
+
+ // These states cannot be mapped; input stream is
+ // off by an event or two
+
+ // case END_OBJECT:
+ // case END_ARRAY:
+ default:
+ }
+ return (JsonNode) ctxt.handleUnexpectedToken(handledType(), p);
+ }
+
+ protected final JsonNode _fromInt(JsonParser p, DeserializationContext ctxt, JsonNodeFactory nodeFactory)
+ throws IOException {
+ JsonParser.NumberType nt;
+ int feats = ctxt.getDeserializationFeatures();
+ if ((feats & F_MASK_INT_COERCIONS) != 0) {
+ if (DeserializationFeature.USE_BIG_INTEGER_FOR_INTS.enabledIn(feats)) {
+ nt = JsonParser.NumberType.BIG_INTEGER;
+ }
+ else if (DeserializationFeature.USE_LONG_FOR_INTS.enabledIn(feats)) {
+ nt = JsonParser.NumberType.LONG;
+ }
+ else {
+ nt = p.getNumberType();
+ }
+ }
+ else {
+ nt = p.getNumberType();
+ }
+ if (nt == JsonParser.NumberType.INT) {
+ return nodeFactory.numberNode(p.getIntValue());
+ }
+ if (nt == JsonParser.NumberType.LONG) {
+ return nodeFactory.numberNode(p.getLongValue());
+ }
+ return nodeFactory.numberNode(p.getBigIntegerValue());
+ }
+
+ protected final JsonNode _fromFloat(JsonParser p, DeserializationContext ctxt, final JsonNodeFactory nodeFactory)
+ throws IOException {
+ JsonParser.NumberType nt = p.getNumberType();
+ if (nt == JsonParser.NumberType.BIG_DECIMAL) {
+ return nodeFactory.numberNode(p.getDecimalValue());
+ }
+ if (ctxt.isEnabled(DeserializationFeature.USE_BIG_DECIMAL_FOR_FLOATS)) {
+ // 20-May-2016, tatu: As per [databind#1028], need to be careful
+ // (note: JDK 1.8 would have `Double.isFinite()`)
+ if (p.isNaN()) {
+ return nodeFactory.numberNode(p.getDoubleValue());
+ }
+ return nodeFactory.numberNode(p.getDecimalValue());
+ }
+ if (nt == JsonParser.NumberType.FLOAT) {
+ return nodeFactory.numberNode(p.getFloatValue());
+ }
+ return nodeFactory.numberNode(p.getDoubleValue());
+ }
+
+ protected final JsonNode _fromEmbedded(JsonParser p, DeserializationContext ctxt, JsonNodeFactory nodeFactory)
+ throws IOException {
+ Object ob = p.getEmbeddedObject();
+ if (ob == null) { // should this occur?
+ return nodeFactory.nullNode();
+ }
+ Class type = ob.getClass();
+ if (type == byte[].class) { // most common special case
+ return nodeFactory.binaryNode((byte[]) ob);
+ }
+ // [databind#743]: Don't forget RawValue
+ if (ob instanceof RawValue) {
+ return nodeFactory.rawValueNode((RawValue) ob);
+ }
+ if (ob instanceof JsonNode) {
+ // [databind#433]: but could also be a JsonNode hiding in there!
+ return (JsonNode) ob;
+ }
+ // any other special handling needed?
+ return nodeFactory.pojoNode(ob);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/ChatFunctionCallDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/ChatFunctionCallDeserializer.java
new file mode 100644
index 0000000..6967435
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/ChatFunctionCallDeserializer.java
@@ -0,0 +1,71 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.ChatFunctionCall;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link ChatFunctionCall} from any JSON
+ * content, using appropriate {@link ChatFunctionCall} type.
+ */
+public class ChatFunctionCallDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ /**
+ * Singleton instance of generic deserializer for
+ * {@link ChatFunctionCallDeserializer}. Only used for types other than JSON Object
+ * and Array.
+ */
+ private final static ChatFunctionCallDeserializer instance = new ChatFunctionCallDeserializer();
+
+ public ChatFunctionCallDeserializer() {
+ // `null` means that explicit "merge" is honored and may or may not work, but
+ // that per-type and global defaults do not enable merging. This because
+ // some node types (Object, Array) do support, others don't.
+ super(ChatFunctionCall.class, null);
+ }
+
+ /**
+ * Factory method for accessing deserializer for specific node type
+ */
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ // For others, generic one works fine
+ return instance;
+ }
+
+ /*
+ * /********************************************************** /* Actual deserializer
+ * implementations /**********************************************************
+ */
+
+ @Override
+ public ChatFunctionCall getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ /**
+ * Implementation that will produce types of any JSON nodes; not just one deserializer
+ * is registered to handle (in case of more specialized handler). Overridden by typed
+ * sub-classes for more thorough checking
+ */
+ @Override
+ public ChatFunctionCall deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new ChatFunctionCall(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/ChatMessageDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/ChatMessageDeserializer.java
new file mode 100644
index 0000000..fcbaec0
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/ChatMessageDeserializer.java
@@ -0,0 +1,70 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.ChatMessage;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link ChatMessage} from any JSON content,
+ * using appropriate {@link ChatMessage} type.
+ */
+public class ChatMessageDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ /**
+ * Singleton instance of generic deserializer for {@link ChatMessageDeserializer}.
+ * Only used for types other than JSON Object and Array.
+ */
+ private final static ChatMessageDeserializer instance = new ChatMessageDeserializer();
+
+ public ChatMessageDeserializer() {
+ // `null` means that explicit "merge" is honored and may or may not work, but
+ // that per-type and global defaults do not enable merging. This because
+ // some node types (Object, Array) do support, others don't.
+ super(ChatMessage.class, null);
+ }
+
+ /**
+ * Factory method for accessing deserializer for specific node type
+ */
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ // For others, generic one works fine
+ return instance;
+ }
+
+ /*
+ * /********************************************************** /* Actual deserializer
+ * implementations /**********************************************************
+ */
+
+ @Override
+ public ChatMessage getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ /**
+ * Implementation that will produce types of any JSON nodes; not just one deserializer
+ * is registered to handle (in case of more specialized handler). Overridden by typed
+ * sub-classes for more thorough checking
+ */
+ @Override
+ public ChatMessage deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new ChatMessage(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/ChoiceDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/ChoiceDeserializer.java
new file mode 100644
index 0000000..f5e22e4
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/ChoiceDeserializer.java
@@ -0,0 +1,70 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.Choice;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link Choice} from any JSON content, using
+ * appropriate {@link Choice} type.
+ */
+public class ChoiceDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ /**
+ * Singleton instance of generic deserializer for {@link ChoiceDeserializer}. Only
+ * used for types other than JSON Object and Array.
+ */
+ private final static ChoiceDeserializer instance = new ChoiceDeserializer();
+
+ public ChoiceDeserializer() {
+ // `null` means that explicit "merge" is honored and may or may not work, but
+ // that per-type and global defaults do not enable merging. This because
+ // some node types (Object, Array) do support, others don't.
+ super(Choice.class, null);
+ }
+
+ /**
+ * Factory method for accessing deserializer for specific node type
+ */
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ // For others, generic one works fine
+ return instance;
+ }
+
+ /*
+ * /********************************************************** /* Actual deserializer
+ * implementations /**********************************************************
+ */
+
+ @Override
+ public Choice getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ /**
+ * Implementation that will produce types of any JSON nodes; not just one deserializer
+ * is registered to handle (in case of more specialized handler). Overridden by typed
+ * sub-classes for more thorough checking
+ */
+ @Override
+ public Choice deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new Choice(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/CodeGeexContextDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/CodeGeexContextDeserializer.java
new file mode 100644
index 0000000..ade7986
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/CodeGeexContextDeserializer.java
@@ -0,0 +1,42 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.params.CodeGeexContext;
+
+import java.io.IOException;
+
+public class CodeGeexContextDeserializer extends BaseNodeDeserializer {
+
+ private static final CodeGeexContextDeserializer instance = new CodeGeexContextDeserializer();
+
+ public CodeGeexContextDeserializer() {
+ super(CodeGeexContext.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public CodeGeexContext getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public CodeGeexContext deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new CodeGeexContext(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/DeltaDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/DeltaDeserializer.java
new file mode 100644
index 0000000..f0585e9
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/DeltaDeserializer.java
@@ -0,0 +1,70 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.Delta;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link Delta} from any JSON content, using
+ * appropriate {@link Delta} type.
+ */
+public class DeltaDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ /**
+ * Singleton instance of generic deserializer for {@link DeltaDeserializer}. Only used
+ * for types other than JSON Object and Array.
+ */
+ private final static DeltaDeserializer instance = new DeltaDeserializer();
+
+ public DeltaDeserializer() {
+ // `null` means that explicit "merge" is honored and may or may not work, but
+ // that per-type and global defaults do not enable merging. This because
+ // some node types (Object, Array) do support, others don't.
+ super(Delta.class, null);
+ }
+
+ /**
+ * Factory method for accessing deserializer for specific node type
+ */
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ // For others, generic one works fine
+ return instance;
+ }
+
+ /*
+ * /********************************************************** /* Actual deserializer
+ * implementations /**********************************************************
+ */
+
+ @Override
+ public Delta getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ /**
+ * Implementation that will produce types of any JSON nodes; not just one deserializer
+ * is registered to handle (in case of more specialized handler). Overridden by typed
+ * sub-classes for more thorough checking
+ */
+ @Override
+ public Delta deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new Delta(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/JsonTypeField.java b/core/src/main/java/ai/z/openapi/service/deserialize/JsonTypeField.java
new file mode 100644
index 0000000..c03c863
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/JsonTypeField.java
@@ -0,0 +1,11 @@
+package ai.z.openapi.service.deserialize;
+
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+
+@Retention(RetentionPolicy.RUNTIME)
+public @interface JsonTypeField {
+
+ String value();
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/JsonTypeMapping.java b/core/src/main/java/ai/z/openapi/service/deserialize/JsonTypeMapping.java
new file mode 100644
index 0000000..70bff0b
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/JsonTypeMapping.java
@@ -0,0 +1,11 @@
+package ai.z.openapi.service.deserialize;
+
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+
+@Retention(RetentionPolicy.RUNTIME)
+public @interface JsonTypeMapping {
+
+ Class[] value();
+
+}
\ No newline at end of file
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/MessageDeserializeFactory.java b/core/src/main/java/ai/z/openapi/service/deserialize/MessageDeserializeFactory.java
new file mode 100644
index 0000000..676d8ad
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/MessageDeserializeFactory.java
@@ -0,0 +1,126 @@
+package ai.z.openapi.service.deserialize;
+
+import ai.z.openapi.service.deserialize.knowledge.document.DocumentDataDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.document.DocumentDataFailInfoDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.document.DocumentFailedInfoDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.document.DocumentObjectDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.document.DocumentPageDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.document.DocumentSuccessInfoDeserializer;
+import ai.z.openapi.service.deserialize.tools.ChoiceDeltaDeserializer;
+import ai.z.openapi.service.deserialize.tools.ChoiceDeltaToolCallDeserializer;
+import ai.z.openapi.service.deserialize.tools.SearchChatMessageDeserializer;
+import ai.z.openapi.service.deserialize.tools.SearchIntentDeserializer;
+import ai.z.openapi.service.deserialize.tools.SearchRecommendDeserializer;
+import ai.z.openapi.service.deserialize.tools.SearchResultDeserializer;
+import ai.z.openapi.service.deserialize.tools.WebSearchChoiceDeserializer;
+import ai.z.openapi.service.deserialize.tools.WebSearchMessageDeserializer;
+import ai.z.openapi.service.deserialize.tools.WebSearchMessageToolCallDeserializer;
+import ai.z.openapi.service.deserialize.tools.WebSearchProDeserializer;
+import ai.z.openapi.service.knowledge.document.DocumentData;
+import ai.z.openapi.service.knowledge.document.DocumentDataFailInfo;
+import ai.z.openapi.service.knowledge.document.DocumentFailedInfo;
+import ai.z.openapi.service.knowledge.document.DocumentObject;
+import ai.z.openapi.service.knowledge.document.DocumentPage;
+import ai.z.openapi.service.knowledge.document.DocumentSuccessInfo;
+import ai.z.openapi.service.model.ChatFunctionCall;
+import ai.z.openapi.service.model.ChatMessage;
+import ai.z.openapi.service.model.Choice;
+import ai.z.openapi.service.model.Delta;
+import ai.z.openapi.service.model.ModelData;
+import ai.z.openapi.service.model.ToolCalls;
+import ai.z.openapi.service.tools.ChoiceDelta;
+import ai.z.openapi.service.tools.ChoiceDeltaToolCall;
+import ai.z.openapi.service.tools.SearchChatMessage;
+import ai.z.openapi.service.tools.SearchIntent;
+import ai.z.openapi.service.tools.SearchRecommend;
+import ai.z.openapi.service.tools.SearchResult;
+import ai.z.openapi.service.tools.WebSearchChoice;
+import ai.z.openapi.service.tools.WebSearchMessage;
+import ai.z.openapi.service.tools.WebSearchMessageToolCall;
+import ai.z.openapi.service.tools.WebSearchPro;
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.PropertyNamingStrategy;
+import com.fasterxml.jackson.databind.module.SimpleModule;
+import ai.z.openapi.service.assistant.AssistantChoice;
+import ai.z.openapi.service.assistant.AssistantCompletion;
+import ai.z.openapi.service.assistant.CompletionUsage;
+import ai.z.openapi.service.assistant.ErrorInfo;
+import ai.z.openapi.service.deserialize.assistant.AssistantChoiceDeserializer;
+import ai.z.openapi.service.deserialize.assistant.AssistantCompletionDeserializer;
+import ai.z.openapi.service.deserialize.assistant.CompletionUsageDeserializer;
+import ai.z.openapi.service.deserialize.assistant.ErrorInfoDeserializer;
+import ai.z.openapi.service.deserialize.embedding.EmbeddingDeserializer;
+import ai.z.openapi.service.deserialize.embedding.EmbeddingResultDeserializer;
+import ai.z.openapi.service.deserialize.image.ImageDeserializer;
+import ai.z.openapi.service.deserialize.image.ImageResultDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.KnowledgeInfoDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.KnowledgePageDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.KnowledgeStatisticsDeserializer;
+import ai.z.openapi.service.deserialize.knowledge.KnowledgeUsedDeserializer;
+import ai.z.openapi.service.deserialize.videos.VideoObjectDeserializer;
+import ai.z.openapi.service.deserialize.videos.VideoResultDeserializer;
+import ai.z.openapi.service.embedding.Embedding;
+import ai.z.openapi.service.embedding.EmbeddingResult;
+import ai.z.openapi.service.image.Image;
+import ai.z.openapi.service.image.ImageResult;
+import ai.z.openapi.service.knowledge.KnowledgeInfo;
+import ai.z.openapi.service.knowledge.KnowledgePage;
+import ai.z.openapi.service.knowledge.KnowledgeStatistics;
+import ai.z.openapi.service.knowledge.KnowledgeUsed;
+import ai.z.openapi.service.model.params.CodeGeexContext;
+import ai.z.openapi.service.videos.VideoObject;
+import ai.z.openapi.service.videos.VideoResult;
+
+public class MessageDeserializeFactory {
+
+ public static ObjectMapper defaultObjectMapper() {
+ ObjectMapper mapper = new ObjectMapper();
+ mapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
+ mapper.setPropertyNamingStrategy(PropertyNamingStrategy.SNAKE_CASE);
+ SimpleModule module = new SimpleModule();
+
+ module.addDeserializer(ModelData.class, new ModelDataDeserializer());
+ module.addDeserializer(Choice.class, new ChoiceDeserializer());
+ module.addDeserializer(ChatMessage.class, new ChatMessageDeserializer());
+ module.addDeserializer(Delta.class, new DeltaDeserializer());
+ module.addDeserializer(ToolCalls.class, new ToolCallsDeserializer());
+ module.addDeserializer(ChatFunctionCall.class, new ChatFunctionCallDeserializer());
+ module.addDeserializer(CodeGeexContext.class, new CodeGeexContextDeserializer());
+ module.addDeserializer(ChoiceDelta.class, new ChoiceDeltaDeserializer());
+ module.addDeserializer(ChoiceDeltaToolCall.class, new ChoiceDeltaToolCallDeserializer());
+ module.addDeserializer(SearchChatMessage.class, new SearchChatMessageDeserializer());
+ module.addDeserializer(SearchIntent.class, new SearchIntentDeserializer());
+ module.addDeserializer(SearchRecommend.class, new SearchRecommendDeserializer());
+ module.addDeserializer(SearchResult.class, new SearchResultDeserializer());
+ module.addDeserializer(WebSearchChoice.class, new WebSearchChoiceDeserializer());
+ module.addDeserializer(WebSearchMessage.class, new WebSearchMessageDeserializer());
+ module.addDeserializer(WebSearchMessageToolCall.class, new WebSearchMessageToolCallDeserializer());
+ module.addDeserializer(WebSearchPro.class, new WebSearchProDeserializer());
+ module.addDeserializer(VideoResult.class, new VideoResultDeserializer());
+ module.addDeserializer(VideoObject.class, new VideoObjectDeserializer());
+ module.addDeserializer(Image.class, new ImageDeserializer());
+ module.addDeserializer(ImageResult.class, new ImageResultDeserializer());
+ module.addDeserializer(KnowledgeInfo.class, new KnowledgeInfoDeserializer());
+ module.addDeserializer(KnowledgeUsed.class, new KnowledgeUsedDeserializer());
+ module.addDeserializer(KnowledgeStatistics.class, new KnowledgeStatisticsDeserializer());
+ module.addDeserializer(KnowledgePage.class, new KnowledgePageDeserializer());
+ module.addDeserializer(DocumentFailedInfo.class, new DocumentFailedInfoDeserializer());
+ module.addDeserializer(DocumentObject.class, new DocumentObjectDeserializer());
+ module.addDeserializer(DocumentSuccessInfo.class, new DocumentSuccessInfoDeserializer());
+ module.addDeserializer(DocumentData.class, new DocumentDataDeserializer());
+ module.addDeserializer(DocumentDataFailInfo.class, new DocumentDataFailInfoDeserializer());
+ module.addDeserializer(DocumentPage.class, new DocumentPageDeserializer());
+ module.addDeserializer(EmbeddingResult.class, new EmbeddingResultDeserializer());
+ module.addDeserializer(Embedding.class, new EmbeddingDeserializer());
+ module.addDeserializer(KnowledgeInfo.class, new KnowledgeInfoDeserializer());
+ module.addDeserializer(AssistantChoice.class, new AssistantChoiceDeserializer());
+ module.addDeserializer(AssistantCompletion.class, new AssistantCompletionDeserializer());
+ module.addDeserializer(CompletionUsage.class, new CompletionUsageDeserializer());
+ module.addDeserializer(ErrorInfo.class, new ErrorInfoDeserializer());
+ mapper.registerModule(module);
+
+ return mapper;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/ModelDataDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/ModelDataDeserializer.java
new file mode 100644
index 0000000..525c077
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/ModelDataDeserializer.java
@@ -0,0 +1,70 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.ModelData;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link ModelData} from any JSON content, using
+ * appropriate {@link ModelData} type.
+ */
+public class ModelDataDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ /**
+ * Singleton instance of generic deserializer for {@link ModelDataDeserializer}. Only
+ * used for types other than JSON Object and Array.
+ */
+ private final static ModelDataDeserializer instance = new ModelDataDeserializer();
+
+ public ModelDataDeserializer() {
+ // `null` means that explicit "merge" is honored and may or may not work, but
+ // that per-type and global defaults do not enable merging. This because
+ // some node types (Object, Array) do support, others don't.
+ super(ModelData.class, null);
+ }
+
+ /**
+ * Factory method for accessing deserializer for specific node type
+ */
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ // For others, generic one works fine
+ return instance;
+ }
+
+ /*
+ * /********************************************************** /* Actual deserializer
+ * implementations /**********************************************************
+ */
+
+ @Override
+ public ModelData getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ /**
+ * Implementation that will produce types of any JSON nodes; not just one deserializer
+ * is registered to handle (in case of more specialized handler). Overridden by typed
+ * sub-classes for more thorough checking
+ */
+ @Override
+ public ModelData deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new ModelData(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/ObjectDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/ObjectDeserializer.java
new file mode 100644
index 0000000..a49d4c8
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/ObjectDeserializer.java
@@ -0,0 +1,54 @@
+package ai.z.openapi.service.deserialize;
+
+import java.io.IOException;
+
+import com.fasterxml.jackson.databind.*;
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonToken;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+public final class ObjectDeserializer extends BaseNodeDeserializer {
+
+ private static final long serialVersionUID = 1L;
+
+ private final static ObjectDeserializer _instance = new ObjectDeserializer();
+
+ private ObjectDeserializer() {
+ super(ObjectNode.class, true);
+ }
+
+ public static ObjectDeserializer getInstance() {
+ return _instance;
+ }
+
+ @Override
+ public ObjectNode deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.isExpectedStartObjectToken()) {
+ return deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ }
+ if (p.hasToken(JsonToken.FIELD_NAME)) {
+ return deserializeObjectAtName(p, ctxt, ctxt.getNodeFactory());
+ }
+ // 23-Sep-2015, tatu: Ugh. We may also be given END_OBJECT (similar to
+ // FIELD_NAME),
+ // if caller has advanced to the first token of Object, but for empty Object
+ if (p.hasToken(JsonToken.END_OBJECT)) {
+ return ctxt.getNodeFactory().objectNode();
+ }
+ return (ObjectNode) ctxt.handleUnexpectedToken(ObjectNode.class, p);
+ }
+
+ /**
+ * Variant needed to support both root-level `updateValue()` and merging.
+ *
+ * @since 2.9
+ */
+ @Override
+ public ObjectNode deserialize(JsonParser p, DeserializationContext ctxt, ObjectNode node) throws IOException {
+ if (p.isExpectedStartObjectToken() || p.hasToken(JsonToken.FIELD_NAME)) {
+ return (ObjectNode) updateObject(p, ctxt, (ObjectNode) node);
+ }
+ return (ObjectNode) ctxt.handleUnexpectedToken(ObjectNode.class, p);
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/ToolCallsDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/ToolCallsDeserializer.java
new file mode 100644
index 0000000..6f4528d
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/ToolCallsDeserializer.java
@@ -0,0 +1,70 @@
+package ai.z.openapi.service.deserialize;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.model.ToolCalls;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link ToolCalls} from any JSON content, using
+ * appropriate {@link ToolCalls} type.
+ */
+public class ToolCallsDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ /**
+ * Singleton instance of generic deserializer for {@link ToolCallsDeserializer}. Only
+ * used for types other than JSON Object and Array.
+ */
+ private final static ToolCallsDeserializer instance = new ToolCallsDeserializer();
+
+ public ToolCallsDeserializer() {
+ // `null` means that explicit "merge" is honored and may or may not work, but
+ // that per-type and global defaults do not enable merging. This because
+ // some node types (Object, Array) do support, others don't.
+ super(ToolCalls.class, null);
+ }
+
+ /**
+ * Factory method for accessing deserializer for specific node type
+ */
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ // For others, generic one works fine
+ return instance;
+ }
+
+ /*
+ * /********************************************************** /* Actual deserializer
+ * implementations /**********************************************************
+ */
+
+ @Override
+ public ToolCalls getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ /**
+ * Implementation that will produce types of any JSON nodes; not just one deserializer
+ * is registered to handle (in case of more specialized handler). Overridden by typed
+ * sub-classes for more thorough checking
+ */
+ @Override
+ public ToolCalls deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new ToolCalls(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/assistant/AssistantChoiceDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/AssistantChoiceDeserializer.java
new file mode 100644
index 0000000..49e8df2
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/AssistantChoiceDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.assistant;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.assistant.AssistantChoice;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link AssistantChoice} from any JSON content,
+ * using appropriate {@link AssistantChoice} type.
+ */
+public class AssistantChoiceDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static AssistantChoiceDeserializer instance = new AssistantChoiceDeserializer();
+
+ public AssistantChoiceDeserializer() {
+ super(AssistantChoice.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public AssistantChoice getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public AssistantChoice deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new AssistantChoice(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/assistant/AssistantCompletionDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/AssistantCompletionDeserializer.java
new file mode 100644
index 0000000..3473cea
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/AssistantCompletionDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.assistant;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.assistant.AssistantCompletion;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link AssistantCompletion} from any JSON
+ * content, using appropriate {@link AssistantCompletion} type.
+ */
+public class AssistantCompletionDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static AssistantCompletionDeserializer instance = new AssistantCompletionDeserializer();
+
+ public AssistantCompletionDeserializer() {
+ super(AssistantCompletion.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public AssistantCompletion getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public AssistantCompletion deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new AssistantCompletion(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/assistant/CompletionUsageDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/CompletionUsageDeserializer.java
new file mode 100644
index 0000000..bcba6ba
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/CompletionUsageDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.assistant;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.assistant.CompletionUsage;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link CompletionUsage} from any JSON content,
+ * using appropriate {@link CompletionUsage} type.
+ */
+public class CompletionUsageDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static CompletionUsageDeserializer instance = new CompletionUsageDeserializer();
+
+ public CompletionUsageDeserializer() {
+ super(CompletionUsage.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public CompletionUsage getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public CompletionUsage deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new CompletionUsage(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/assistant/ErrorInfoDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/ErrorInfoDeserializer.java
new file mode 100644
index 0000000..52eebb2
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/ErrorInfoDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.assistant;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.assistant.ErrorInfo;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link ErrorInfo} from any JSON content, using
+ * appropriate {@link ErrorInfo} type.
+ */
+public class ErrorInfoDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static ErrorInfoDeserializer instance = new ErrorInfoDeserializer();
+
+ public ErrorInfoDeserializer() {
+ super(ErrorInfo.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public ErrorInfo getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public ErrorInfo deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new ErrorInfo(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/assistant/message/MessageContentDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/message/MessageContentDeserializer.java
new file mode 100644
index 0000000..f3076b2
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/message/MessageContentDeserializer.java
@@ -0,0 +1,140 @@
+package ai.z.openapi.service.deserialize.assistant.message;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.databind.*;
+import ai.z.openapi.service.assistant.message.MessageContent;
+import ai.z.openapi.service.assistant.message.TextContentBlock;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+import ai.z.openapi.service.deserialize.JsonTypeMapping;
+
+import java.io.IOException;
+import java.lang.reflect.Field;
+import java.lang.reflect.ParameterizedType;
+import java.util.List;
+
+public class MessageContentDeserializer extends JsonDeserializer {
+
+ @Override
+ public MessageContent deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ JsonNode node = p.readValueAsTree();
+
+ // Get JsonTypeMapping annotation from MessageContent class
+ JsonTypeMapping mapping = MessageContent.class.getAnnotation(JsonTypeMapping.class);
+ if (mapping == null) {
+ throw new IllegalStateException("Missing JsonTypeMapping annotation on MessageContent class");
+ }
+
+ // Iterate through classes defined in annotation to determine suitable class based
+ // on annotation or static method values
+ for (Class clazz : mapping.value()) {
+ JsonTypeField typeField = clazz.getAnnotation(JsonTypeField.class);
+
+ if (typeField != null && node.has(typeField.value())) {
+ try {
+ // Create instance of the class
+ Object obj = clazz.getDeclaredConstructor().newInstance();
+
+ // Use reflection to manually set field values
+ for (Field field : clazz.getDeclaredFields()) {
+ field.setAccessible(true);
+ // field.getName() gets the value from JsonProperty annotation
+ // above
+ JsonProperty annotation = field.getAnnotation(JsonProperty.class);
+ String name = null;
+ if (annotation == null) {
+ name = field.getName();
+ }
+ else {
+ name = annotation.value();
+ }
+ if (node.has(name)) {
+ // Set values based on field type, assuming fields are
+ // primitive types or strings
+ if (field.getType().equals(String.class)) {
+ field.set(obj, node.get(name).asText());
+ }
+ else if (field.getType().equals(int.class) || field.getType().equals(Integer.class)) {
+ field.set(obj, node.get(name).asInt());
+ }
+ else if (field.getType().equals(boolean.class) || field.getType().equals(Boolean.class)) {
+ field.set(obj, node.get(name).asBoolean());
+ }
+ else if (List.class.isAssignableFrom(field.getType())) {
+ // Check if generic type is ToolsType
+ ParameterizedType listType = (ParameterizedType) field.getGenericType();
+ Class listGenericType = (Class) listType.getActualTypeArguments()[0];
+
+ if (listGenericType.equals(ToolsType.class)) {
+ // Deserialize to List
+ List list = new ObjectMapper().readerForListOf(ToolsType.class)
+ .readValue(node.get(name));
+ field.set(obj, list);
+ }
+ else {
+ throw new IllegalArgumentException(
+ "Unsupported generic list type: " + listGenericType);
+ }
+ }
+ else {
+ throw new IllegalArgumentException("Unsupported field type: " + field.getType());
+ }
+
+ }
+ }
+
+ return (MessageContent) obj;
+ }
+ catch (Exception e) {
+ throw new RuntimeException("Error while creating instance of " + clazz.getName(), e);
+ }
+ }
+ }
+
+ // Create instance of the class
+ try {
+ MessageContent obj = TextContentBlock.class.getDeclaredConstructor().newInstance();
+
+ // Use reflection to manually set field values
+ for (Field field : TextContentBlock.class.getDeclaredFields()) {
+ field.setAccessible(true);
+ // field.getName() gets the value from JsonProperty annotation above
+ JsonProperty annotation = field.getAnnotation(JsonProperty.class);
+ String name = null;
+ if (annotation == null) {
+ name = field.getName();
+ }
+ else {
+ name = annotation.value();
+ }
+ if (node.has(name)) {
+ // Set values based on field type, assuming fields are primitive types
+ // or strings
+ if (field.getType().equals(String.class)) {
+ field.set(obj, node.get(name).asText());
+ }
+ else if (field.getType().equals(int.class) || field.getType().equals(Integer.class)) {
+ field.set(obj, node.get(name).asInt());
+ }
+ else if (field.getType().equals(boolean.class) || field.getType().equals(Boolean.class)) {
+ field.set(obj, node.get(name).asBoolean());
+
+ }
+ else {
+ // For other types, use ObjectMapper for direct conversion
+ Object o = new ObjectMapper().treeToValue(node.get(name), field.getType());
+ field.set(obj, o);
+ }
+ }
+ }
+
+ return obj;
+ }
+ catch (Exception e) {
+ throw new RuntimeException("Error while creating instance of " + TextContentBlock.class.getName(), e);
+ }
+
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/assistant/message/tools/ToolsTypeDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/message/tools/ToolsTypeDeserializer.java
new file mode 100644
index 0000000..d82b58d
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/assistant/message/tools/ToolsTypeDeserializer.java
@@ -0,0 +1,80 @@
+package ai.z.openapi.service.deserialize.assistant.message.tools;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.databind.*;
+import ai.z.openapi.service.assistant.message.tools.ToolsType;
+import ai.z.openapi.service.deserialize.JsonTypeField;
+import ai.z.openapi.service.deserialize.JsonTypeMapping;
+
+import java.io.IOException;
+import java.lang.reflect.Field;
+
+public class ToolsTypeDeserializer extends JsonDeserializer {
+
+ @Override
+ public ToolsType deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ JsonNode node = p.readValueAsTree();
+
+ // Get JsonTypeMapping annotation from MessageContent class
+ JsonTypeMapping mapping = ToolsType.class.getAnnotation(JsonTypeMapping.class);
+ if (mapping == null) {
+ throw new IllegalStateException("Missing JsonTypeMapping annotation on MessageContent class");
+ }
+
+ // Iterate through classes defined in annotation to determine suitable class based
+ // on annotation or static method values
+ for (Class clazz : mapping.value()) {
+ JsonTypeField typeField = clazz.getAnnotation(JsonTypeField.class);
+
+ if (typeField != null && node.has(typeField.value())) {
+ try {
+ // Create instance of the class
+ Object obj = clazz.getDeclaredConstructor().newInstance();
+
+ // Use reflection to manually set field values
+ for (Field field : clazz.getDeclaredFields()) {
+ field.setAccessible(true);
+ // field.getName() gets the value from JsonProperty annotation
+ // above
+ JsonProperty annotation = field.getAnnotation(JsonProperty.class);
+ String name = null;
+ if (annotation == null) {
+ name = field.getName();
+ }
+ else {
+ name = annotation.value();
+ }
+ if (node.has(name)) {
+ // Set values based on field type, assuming fields are
+ // primitive types or strings
+ if (field.getType().equals(String.class)) {
+ field.set(obj, node.get(name).asText());
+ }
+ else if (field.getType().equals(int.class) || field.getType().equals(Integer.class)) {
+ field.set(obj, node.get(name).asInt());
+ }
+ else if (field.getType().equals(boolean.class) || field.getType().equals(Boolean.class)) {
+ field.set(obj, node.get(name).asBoolean());
+
+ }
+ else {
+ // For other types, use ObjectMapper for direct conversion
+ Object o = new ObjectMapper().treeToValue(node.get(name), field.getType());
+ field.set(obj, o);
+ }
+ }
+ }
+
+ return (ToolsType) obj;
+ }
+ catch (Exception e) {
+ throw new RuntimeException("Error while creating instance of " + clazz.getName(), e);
+ }
+ }
+ }
+
+ throw new IllegalArgumentException("Cannot determine type for JSON: " + node.toString());
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/embedding/EmbeddingDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/embedding/EmbeddingDeserializer.java
new file mode 100644
index 0000000..f757b77
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/embedding/EmbeddingDeserializer.java
@@ -0,0 +1,52 @@
+
+package ai.z.openapi.service.deserialize.embedding;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.embedding.Embedding;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link Embedding} from any JSON content, using
+ * appropriate {@link Embedding} type.
+ */
+public class EmbeddingDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static EmbeddingDeserializer instance = new EmbeddingDeserializer();
+
+ public EmbeddingDeserializer() {
+ super(Embedding.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public Embedding getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public Embedding deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new Embedding(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/embedding/EmbeddingResultDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/embedding/EmbeddingResultDeserializer.java
new file mode 100644
index 0000000..dd5ad82
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/embedding/EmbeddingResultDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.embedding;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.embedding.EmbeddingResult;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link EmbeddingResult} from any JSON content,
+ * using appropriate {@link EmbeddingResult} type.
+ */
+public class EmbeddingResultDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static EmbeddingResultDeserializer instance = new EmbeddingResultDeserializer();
+
+ public EmbeddingResultDeserializer() {
+ super(EmbeddingResult.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public EmbeddingResult getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public EmbeddingResult deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new EmbeddingResult(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/image/ImageDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/image/ImageDeserializer.java
new file mode 100644
index 0000000..7cd15e7
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/image/ImageDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.image;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.image.Image;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link Image} from any JSON content, using
+ * appropriate {@link Image} type.
+ */
+public class ImageDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static ImageDeserializer instance = new ImageDeserializer();
+
+ public ImageDeserializer() {
+ super(Image.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public Image getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public Image deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new Image(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/image/ImageResultDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/image/ImageResultDeserializer.java
new file mode 100644
index 0000000..583166e
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/image/ImageResultDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.image;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.image.ImageResult;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link ImageResult} from any JSON content,
+ * using appropriate {@link ImageResult} type.
+ */
+public class ImageResultDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static ImageResultDeserializer instance = new ImageResultDeserializer();
+
+ public ImageResultDeserializer() {
+ super(ImageResult.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public ImageResult getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public ImageResult deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new ImageResult(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeInfoDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeInfoDeserializer.java
new file mode 100644
index 0000000..fd00a3f
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeInfoDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.knowledge;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.knowledge.KnowledgeInfo;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link KnowledgeInfo} from any JSON content,
+ * using appropriate {@link KnowledgeInfo} type.
+ */
+public class KnowledgeInfoDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static KnowledgeInfoDeserializer instance = new KnowledgeInfoDeserializer();
+
+ public KnowledgeInfoDeserializer() {
+ super(KnowledgeInfo.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public KnowledgeInfo getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public KnowledgeInfo deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new KnowledgeInfo(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgePageDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgePageDeserializer.java
new file mode 100644
index 0000000..e64a669
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgePageDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.knowledge;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.knowledge.KnowledgePage;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link KnowledgePage} from any JSON content,
+ * using appropriate {@link KnowledgePage} type.
+ */
+public class KnowledgePageDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static KnowledgePageDeserializer instance = new KnowledgePageDeserializer();
+
+ public KnowledgePageDeserializer() {
+ super(KnowledgePage.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public KnowledgePage getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public KnowledgePage deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new KnowledgePage(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeStatisticsDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeStatisticsDeserializer.java
new file mode 100644
index 0000000..f741a84
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeStatisticsDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.knowledge;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.knowledge.KnowledgeStatistics;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link KnowledgeStatistics} from any JSON
+ * content, using appropriate {@link KnowledgeStatistics} type.
+ */
+public class KnowledgeStatisticsDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static KnowledgeStatisticsDeserializer instance = new KnowledgeStatisticsDeserializer();
+
+ public KnowledgeStatisticsDeserializer() {
+ super(KnowledgeStatistics.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public KnowledgeStatistics getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public KnowledgeStatistics deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new KnowledgeStatistics(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeUsedDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeUsedDeserializer.java
new file mode 100644
index 0000000..b64cfc5
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/KnowledgeUsedDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.knowledge;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.knowledge.KnowledgeUsed;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link KnowledgeUsed} from any JSON content,
+ * using appropriate {@link KnowledgeUsed} type.
+ */
+public class KnowledgeUsedDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static KnowledgeUsedDeserializer instance = new KnowledgeUsedDeserializer();
+
+ public KnowledgeUsedDeserializer() {
+ super(KnowledgeUsed.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public KnowledgeUsed getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public KnowledgeUsed deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new KnowledgeUsed(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/document/DocumentDataDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/document/DocumentDataDeserializer.java
new file mode 100644
index 0000000..7dd4f80
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/document/DocumentDataDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.knowledge.document;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.knowledge.document.DocumentData;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link DocumentData} from any JSON content,
+ * using appropriate {@link DocumentData} type.
+ */
+public class DocumentDataDeserializer extends BaseNodeDeserializer {
+
+ private final static ObjectMapper MAPPER = new ObjectMapper();
+
+ private final static DocumentDataDeserializer instance = new DocumentDataDeserializer();
+
+ public DocumentDataDeserializer() {
+ super(DocumentData.class, null);
+ }
+
+ public static JsonDeserializer getDeserializer(Class nodeClass) {
+ if (nodeClass == ObjectNode.class) {
+ return ObjectDeserializer.getInstance();
+ }
+ return instance;
+ }
+
+ @Override
+ public DocumentData getNullValue(DeserializationContext ctxt) {
+ return null;
+ }
+
+ @Override
+ public DocumentData deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
+ if (p.currentTokenId() == JsonTokenId.ID_START_OBJECT) {
+ ObjectNode jsonNodes = deserializeObject(p, ctxt, ctxt.getNodeFactory());
+ return new DocumentData(jsonNodes);
+ }
+ return null;
+ }
+
+}
diff --git a/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/document/DocumentDataFailInfoDeserializer.java b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/document/DocumentDataFailInfoDeserializer.java
new file mode 100644
index 0000000..adf3ffc
--- /dev/null
+++ b/core/src/main/java/ai/z/openapi/service/deserialize/knowledge/document/DocumentDataFailInfoDeserializer.java
@@ -0,0 +1,51 @@
+package ai.z.openapi.service.deserialize.knowledge.document;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonTokenId;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import ai.z.openapi.service.deserialize.BaseNodeDeserializer;
+import ai.z.openapi.service.deserialize.ObjectDeserializer;
+import ai.z.openapi.service.knowledge.document.DocumentDataFailInfo;
+
+import java.io.IOException;
+
+/**
+ * Deserializer that can build instances of {@link DocumentDataFailInfo} from any JSON
+ * content, using appropriate {@link DocumentDataFailInfo} type.
+ */
+public class DocumentDataFailInfoDeserializer extends BaseNodeDeserializer