Commit 13684eee authored by 林洋洋's avatar 林洋洋

OPenAI 实现创建chatmodel

parent 5f45381d
...@@ -2,8 +2,13 @@ package com.ask.model; ...@@ -2,8 +2,13 @@ package com.ask.model;
import com.ask.api.entity.AskModel; import com.ask.api.entity.AskModel;
import org.springframework.ai.chat.model.ChatModel; import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.deepseek.DeepSeekChatModel;
import org.springframework.ai.deepseek.DeepSeekChatOptions;
import org.springframework.ai.deepseek.api.DeepSeekApi;
import org.springframework.ai.document.MetadataMode; import org.springframework.ai.document.MetadataMode;
import org.springframework.ai.embedding.EmbeddingModel; import org.springframework.ai.embedding.EmbeddingModel;
import org.springframework.ai.openai.OpenAiChatModel;
import org.springframework.ai.openai.OpenAiChatOptions;
import org.springframework.ai.openai.OpenAiEmbeddingModel; import org.springframework.ai.openai.OpenAiEmbeddingModel;
import org.springframework.ai.openai.OpenAiEmbeddingOptions; import org.springframework.ai.openai.OpenAiEmbeddingOptions;
import org.springframework.ai.openai.api.OpenAiApi; import org.springframework.ai.openai.api.OpenAiApi;
...@@ -12,7 +17,10 @@ import org.springframework.ai.retry.RetryUtils; ...@@ -12,7 +17,10 @@ import org.springframework.ai.retry.RetryUtils;
public class IOpenAiModel implements IBaseModel { public class IOpenAiModel implements IBaseModel {
@Override @Override
public ChatModel buildChatModel(AskModel askModel) { public ChatModel buildChatModel(AskModel askModel) {
return null; return OpenAiChatModel.builder()
.openAiApi(OpenAiApi.builder().baseUrl(askModel.getBaseUrl()).apiKey(askModel.getKey()).build())
.defaultOptions(OpenAiChatOptions.builder().model(askModel.getModelName()).temperature(Double.valueOf(askModel.getTemperature())).maxTokens(askModel.getMaxTokens()).build())
.build();
} }
@Override @Override
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment