From 278fd39195bb9327733f64b89e71dcf2b39d78dc Mon Sep 17 00:00:00 2001
From: Maple Gao <esanisa@gmail.com>
Date: Mon, 1 Apr 2024 14:33:58 +0800
Subject: [PATCH] feat: add Claude TopK

---
 relay/channel/claude/relay-claude.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/relay/channel/claude/relay-claude.go b/relay/channel/claude/relay-claude.go
index 4de8dc0..2b5d3d2 100644
--- a/relay/channel/claude/relay-claude.go
+++ b/relay/channel/claude/relay-claude.go
@@ -34,6 +34,7 @@ func requestOpenAI2ClaudeComplete(textRequest dto.GeneralOpenAIRequest) *ClaudeR
 		StopSequences:     nil,
 		Temperature:       textRequest.Temperature,
 		TopP:              textRequest.TopP,
+		TopK:              textRequest.TopK,
 		Stream:            textRequest.Stream,
 	}
 	if claudeRequest.MaxTokensToSample == 0 {
@@ -63,6 +64,7 @@ func requestOpenAI2ClaudeMessage(textRequest dto.GeneralOpenAIRequest) (*ClaudeR
 		StopSequences: nil,
 		Temperature:   textRequest.Temperature,
 		TopP:          textRequest.TopP,
+		TopK:          textRequest.TopK,
 		Stream:        textRequest.Stream,
 	}
 	if claudeRequest.MaxTokens == 0 {