@@ -2,6 +2,7 @@ package io.cequence.openaiscala.anthropic.service.impl
22
33import akka .actor .ActorSystem
44import akka .stream .Materializer
5+ import io .cequence .openaiscala .anthropic .domain .Content .SingleString
56import io .cequence .openaiscala .anthropic .domain .Message .UserMessage
67import io .cequence .openaiscala .anthropic .domain .settings .AnthropicCreateMessageSettings
78import 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