Skip to content

Commit 63d142a

Browse files
stainless-app[bot]stainless-app[bot]
committed
fix(tests): add missing query/header params
1 parent 8edb811 commit 63d142a

1 file changed

Lines changed: 67 additions & 51 deletions

File tree

‎arcade-java-core/src/test/kotlin/dev/arcade/services/ServiceParamsTest.kt‎

Lines changed: 67 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,9 @@ import dev.arcade.client.okhttp.ArcadeOkHttpClient
1616
import dev.arcade.core.JsonValue
1717
import dev.arcade.models.chat.ChatMessage
1818
import dev.arcade.models.chat.ChatRequest
19+
import dev.arcade.models.chat.completions.CompletionCreateParams
1920
import dev.arcade.models.tools.ExecuteToolRequest
21+
import dev.arcade.models.tools.ToolExecuteParams
2022
import org.junit.jupiter.api.BeforeEach
2123
import org.junit.jupiter.api.Test
2224
import org.junit.jupiter.api.parallel.ResourceLock
@@ -42,54 +44,62 @@ internal class ServiceParamsTest {
4244
stubFor(post(anyUrl()).willReturn(ok("{}")))
4345

4446
completionService.create(
45-
ChatRequest.builder()
46-
.frequencyPenalty(0.0)
47-
.logitBias(
48-
ChatRequest.LogitBias.builder()
49-
.putAdditionalProperty("foo", JsonValue.from(0))
50-
.build()
51-
)
52-
.logprobs(true)
53-
.maxTokens(0L)
54-
.addMessage(
55-
ChatMessage.builder()
56-
.content("content")
57-
.role("role")
58-
.name("name")
59-
.toolCallId("tool_call_id")
60-
.addToolCall(
61-
ChatMessage.ToolCall.builder()
62-
.id("id")
63-
.function(
64-
ChatMessage.ToolCall.Function.builder()
65-
.arguments("arguments")
66-
.name("name")
47+
CompletionCreateParams.builder()
48+
.chatRequest(
49+
ChatRequest.builder()
50+
.frequencyPenalty(0.0)
51+
.logitBias(
52+
ChatRequest.LogitBias.builder()
53+
.putAdditionalProperty("foo", JsonValue.from(0))
54+
.build()
55+
)
56+
.logprobs(true)
57+
.maxTokens(0L)
58+
.addMessage(
59+
ChatMessage.builder()
60+
.content("content")
61+
.role("role")
62+
.name("name")
63+
.toolCallId("tool_call_id")
64+
.addToolCall(
65+
ChatMessage.ToolCall.builder()
66+
.id("id")
67+
.function(
68+
ChatMessage.ToolCall.Function.builder()
69+
.arguments("arguments")
70+
.name("name")
71+
.build()
72+
)
73+
.type(ChatMessage.ToolCall.Type.FUNCTION)
6774
.build()
6875
)
69-
.type(ChatMessage.ToolCall.Type.FUNCTION)
7076
.build()
7177
)
78+
.model("model")
79+
.n(0L)
80+
.parallelToolCalls(true)
81+
.presencePenalty(0.0)
82+
.responseFormat(
83+
ChatRequest.ResponseFormat.builder()
84+
.type(ChatRequest.ResponseFormat.Type.JSON_OBJECT)
85+
.build()
86+
)
87+
.seed(0L)
88+
.addStop("string")
89+
.stream(true)
90+
.streamOptions(
91+
ChatRequest.StreamOptions.builder().includeUsage(true).build()
92+
)
93+
.temperature(0.0)
94+
.toolChoice(JsonValue.from(mapOf<String, Any>()))
95+
.tools(JsonValue.from(mapOf<String, Any>()))
96+
.topLogprobs(0L)
97+
.topP(0.0)
98+
.user("user")
7299
.build()
73100
)
74-
.model("model")
75-
.n(0L)
76-
.parallelToolCalls(true)
77-
.presencePenalty(0.0)
78-
.responseFormat(
79-
ChatRequest.ResponseFormat.builder()
80-
.type(ChatRequest.ResponseFormat.Type.JSON_OBJECT)
81-
.build()
82-
)
83-
.seed(0L)
84-
.addStop("string")
85-
.stream(true)
86-
.streamOptions(ChatRequest.StreamOptions.builder().includeUsage(true).build())
87-
.temperature(0.0)
88-
.toolChoice(JsonValue.from(mapOf<String, Any>()))
89-
.tools(JsonValue.from(mapOf<String, Any>()))
90-
.topLogprobs(0L)
91-
.topP(0.0)
92-
.user("user")
101+
.putAdditionalHeader("Secret-Header", "42")
102+
.putAdditionalQueryParam("secret_query_param", "42")
93103
.build()
94104
)
95105

@@ -106,17 +116,23 @@ internal class ServiceParamsTest {
106116
stubFor(post(anyUrl()).willReturn(ok("{}")))
107117

108118
toolService.execute(
109-
ExecuteToolRequest.builder()
110-
.toolName("tool_name")
111-
.includeErrorStacktrace(true)
112-
.input(
113-
ExecuteToolRequest.Input.builder()
114-
.putAdditionalProperty("foo", JsonValue.from("bar"))
119+
ToolExecuteParams.builder()
120+
.executeToolRequest(
121+
ExecuteToolRequest.builder()
122+
.toolName("tool_name")
123+
.includeErrorStacktrace(true)
124+
.input(
125+
ExecuteToolRequest.Input.builder()
126+
.putAdditionalProperty("foo", JsonValue.from("bar"))
127+
.build()
128+
)
129+
.runAt("run_at")
130+
.toolVersion("tool_version")
131+
.userId("user_id")
115132
.build()
116133
)
117-
.runAt("run_at")
118-
.toolVersion("tool_version")
119-
.userId("user_id")
134+
.putAdditionalHeader("Secret-Header", "42")
135+
.putAdditionalQueryParam("secret_query_param", "42")
120136
.build()
121137
)
122138

0 commit comments

Comments
 (0)