Skip to content

Commit 11f34d8

Browse files
committed
Anthropic test - fix
1 parent 8158941 commit 11f34d8

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

anthropic-client/src/test/scala/io/cequence/openaiscala/anthropic/service/impl/AnthropicServiceSpec.scala

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package io.cequence.openaiscala.anthropic.service.impl
22

33
import akka.actor.ActorSystem
44
import akka.stream.Materializer
5+
import io.cequence.openaiscala.anthropic.domain.Content.SingleString
56
import io.cequence.openaiscala.anthropic.domain.Message.UserMessage
67
import io.cequence.openaiscala.anthropic.domain.settings.AnthropicCreateMessageSettings
78
import io.cequence.openaiscala.anthropic.service._
@@ -17,6 +18,7 @@ class AnthropicServiceSpec extends AsyncWordSpec with GivenWhenThen {
1718
implicit val ec: ExecutionContext = ExecutionContext.global
1819
implicit val materializer: Materializer = Materializer(ActorSystem())
1920

21+
private val role = SingleString("You are a helpful assistant.")
2022
private val irrelevantMessages = Seq(UserMessage("Hello"))
2123
private val settings = AnthropicCreateMessageSettings(
2224
NonOpenAIModelId.claude_3_haiku_20240307,
@@ -27,25 +29,25 @@ class AnthropicServiceSpec extends AsyncWordSpec with GivenWhenThen {
2729

2830
"should throw AnthropicScalaUnauthorizedException when 401" ignore {
2931
recoverToSucceededIf[AnthropicScalaUnauthorizedException] {
30-
TestFactory.mockedService401().createMessage(irrelevantMessages, None, settings)
32+
TestFactory.mockedService401().createMessage(Some(role), irrelevantMessages, settings)
3133
}
3234
}
3335

3436
"should throw AnthropicScalaUnauthorizedException when 403" ignore {
3537
recoverToSucceededIf[AnthropicScalaUnauthorizedException] {
36-
TestFactory.mockedService403().createMessage(irrelevantMessages, None, settings)
38+
TestFactory.mockedService403().createMessage(Some(role), irrelevantMessages, settings)
3739
}
3840
}
3941

4042
"should throw AnthropicScalaNotFoundException when 404" ignore {
4143
recoverToSucceededIf[AnthropicScalaNotFoundException] {
42-
TestFactory.mockedService404().createMessage(irrelevantMessages, None, settings)
44+
TestFactory.mockedService404().createMessage(Some(role), irrelevantMessages, settings)
4345
}
4446
}
4547

4648
"should throw AnthropicScalaNotFoundException when 429" ignore {
4749
recoverToSucceededIf[AnthropicScalaRateLimitException] {
48-
TestFactory.mockedService429().createMessage(irrelevantMessages, None, settings)
50+
TestFactory.mockedService429().createMessage(Some(role), irrelevantMessages, settings)
4951
}
5052
}
5153

0 commit comments

Comments
 (0)