20250329 通过application配置apikey以及model

This commit is contained in:
liangjinglin 2025-03-29 23:54:35 +08:00
parent 4fbeca899b
commit ee3f64a5c6
2 changed files with 14 additions and 3 deletions

View File

@ -13,6 +13,7 @@ import dev.langchain4j.service.AiServices;
import dev.langchain4j.store.embedding.EmbeddingStoreIngestor;
import dev.langchain4j.store.embedding.inmemory.InMemoryEmbeddingStore;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@ -24,11 +25,17 @@ public class AssistantInit {
// All files in a directory, txt seems to be faster
@Value("${langchain4j.api-key}")
private String apiKey;
@Value("${langchain4j.model}")
private String model;
@Bean
public Assist init() {
ChatLanguageModel qwenModel = QwenChatModel.builder()
.apiKey("sk-2f703a41fff0488e9b6888013d2ee58a")
.modelName("deepseek-v3")
.apiKey(apiKey)
.modelName(model)
.build();
List<Document> documents = FileSystemDocumentLoader.loadDocuments("E:\\ideaProject\\liang-ai");
// for simplicity, we will use an in-memory one:
@ -38,7 +45,7 @@ public class AssistantInit {
.chatMemoryProvider(memoryId -> MessageWindowChatMemory.withMaxMessages(10))
.chatLanguageModel(qwenModel)
.contentRetriever(EmbeddingStoreContentRetriever.from(embeddingStore))
.tools(new MyCalculator())
// .tools(new MyCalculator())
.build();
}
}

View File

@ -2,3 +2,7 @@ server.port=8080
langchain4j.community.dashscope.chat-model.api-key=sk-2f703a41fff0488e9b6888013d2ee58a
langchain4j.community.dashscope.chat-model.model-name=deepseek-v3
langchain4j.api-key=sk-2f703a41fff0488e9b6888013d2ee58a
langchain4j.model=deepseek-v3