Skip to content

Commit 2ad6f43

Browse files
committed
fix: fix axure auto service issue
1 parent 53a192b commit 2ad6f43

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/main/kotlin/cc/unitmesh/devti/llms/azure/AzureOpenAIProvider.kt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ class AzureOpenAIProvider(val project: Project) : LLMProvider {
5050
private val timeout = Duration.ofSeconds(600)
5151
private var client = OkHttpClient().newBuilder().readTimeout(timeout).build()
5252
private val openAiVersion: String
53+
private val maxTokenLength: Int
54+
get() = AutoDevSettingsState.getInstance().fetchMaxTokenLength()
55+
5356

5457
init {
5558
val prompts = autoDevSettingsState.customPrompts
@@ -78,9 +81,10 @@ class AzureOpenAIProvider(val project: Project) : LLMProvider {
7881
fun prompt(instruction: String, input: String): String {
7982
val promptText = "$instruction\n$input"
8083
val systemMessage = ChatMessage(ChatMessageRole.USER.value(), promptText)
81-
if (historyMessageLength > 8192) {
84+
if (historyMessageLength > maxTokenLength) {
8285
messages.clear()
8386
}
87+
8488
messages.add(SimpleOpenAIFormat.fromChatMessage(systemMessage))
8589
val requestText = Json.encodeToString<SimpleOpenAIBody>(
8690
SimpleOpenAIBody(

src/main/kotlin/cc/unitmesh/devti/llms/openai/OpenAIProvider.kt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ class OpenAIProvider(val project: Project) : LLMProvider {
126126
}
127127

128128
messages.add(systemMessage)
129+
logger.info("messages length: ${messages.size}")
129130

130131
return ChatCompletionRequest.builder()
131132
.model(openAiVersion)

0 commit comments

Comments
 (0)