From 57a79bd463d2854e64082d338dfb5e507576b1c3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=82=96=E6=AC=A2?= <296795054@qq.com>
Date: Sun, 28 Apr 2024 18:06:13 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=9F=A5=E8=AF=86=E5=BA=93?=
=?UTF-8?q?=E6=90=9C=E7=B4=A2=E6=B5=8B=E8=AF=95ChatModel=E4=B8=BA=E7=A9=BA?=
=?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
src/Service/FastWiki.Service/Options/OpenAIOption.cs | 5 +++++
src/Service/FastWiki.Service/Service/WikiMemoryService.cs | 2 +-
2 files changed, 6 insertions(+), 1 deletion(-)
diff --git a/src/Service/FastWiki.Service/Options/OpenAIOption.cs b/src/Service/FastWiki.Service/Options/OpenAIOption.cs
index d010b4f..c834f9c 100644
--- a/src/Service/FastWiki.Service/Options/OpenAIOption.cs
+++ b/src/Service/FastWiki.Service/Options/OpenAIOption.cs
@@ -14,6 +14,11 @@ public class OpenAIOption
/// </summary>
public static string EmbeddingEndpoint { get; set; }
+ /// <summary>
+ /// 对话模型的 ChatModel
+ /// </summary>
+ public static string ChatModel { get; set; }
+
/// <summary>
/// 对话模型的 API Key
/// </summary>
diff --git a/src/Service/FastWiki.Service/Service/WikiMemoryService.cs b/src/Service/FastWiki.Service/Service/WikiMemoryService.cs
index 8e3adc4..6ec40b7 100644
--- a/src/Service/FastWiki.Service/Service/WikiMemoryService.cs
+++ b/src/Service/FastWiki.Service/Service/WikiMemoryService.cs
@@ -147,7 +147,7 @@ public sealed class WikiMemoryService : ISingletonDependency
.WithOpenAITextGeneration(new OpenAIConfig()
{
APIKey = OpenAIOption.ChatToken,
- TextModel = model ?? string.Empty
+ TextModel = model ?? OpenAIOption.ChatModel
}, null, new HttpClient(HttpClientHandler))
.WithOpenAITextEmbeddingGeneration(new OpenAIConfig()
{
--
Gitee