@@ -2,6 +2,7 @@ package io.cequence.openaiscala.anthropic.service.impl
2
2
3
3
import akka .actor .ActorSystem
4
4
import akka .stream .Materializer
5
+ import io .cequence .openaiscala .anthropic .domain .Content .SingleString
5
6
import io .cequence .openaiscala .anthropic .domain .Message .UserMessage
6
7
import io .cequence .openaiscala .anthropic .domain .settings .AnthropicCreateMessageSettings
7
8
import io .cequence .openaiscala .anthropic .service ._
@@ -17,6 +18,7 @@ class AnthropicServiceSpec extends AsyncWordSpec with GivenWhenThen {
17
18
implicit val ec : ExecutionContext = ExecutionContext .global
18
19
implicit val materializer : Materializer = Materializer (ActorSystem ())
19
20
21
+ private val role = SingleString (" You are a helpful assistant." )
20
22
private val irrelevantMessages = Seq (UserMessage (" Hello" ))
21
23
private val settings = AnthropicCreateMessageSettings (
22
24
NonOpenAIModelId .claude_3_haiku_20240307,
@@ -27,25 +29,25 @@ class AnthropicServiceSpec extends AsyncWordSpec with GivenWhenThen {
27
29
28
30
" should throw AnthropicScalaUnauthorizedException when 401" ignore {
29
31
recoverToSucceededIf[AnthropicScalaUnauthorizedException ] {
30
- TestFactory .mockedService401().createMessage(irrelevantMessages, None , settings)
32
+ TestFactory .mockedService401().createMessage(Some (role), irrelevantMessages , settings)
31
33
}
32
34
}
33
35
34
36
" should throw AnthropicScalaUnauthorizedException when 403" ignore {
35
37
recoverToSucceededIf[AnthropicScalaUnauthorizedException ] {
36
- TestFactory .mockedService403().createMessage(irrelevantMessages, None , settings)
38
+ TestFactory .mockedService403().createMessage(Some (role), irrelevantMessages , settings)
37
39
}
38
40
}
39
41
40
42
" should throw AnthropicScalaNotFoundException when 404" ignore {
41
43
recoverToSucceededIf[AnthropicScalaNotFoundException ] {
42
- TestFactory .mockedService404().createMessage(irrelevantMessages, None , settings)
44
+ TestFactory .mockedService404().createMessage(Some (role), irrelevantMessages , settings)
43
45
}
44
46
}
45
47
46
48
" should throw AnthropicScalaNotFoundException when 429" ignore {
47
49
recoverToSucceededIf[AnthropicScalaRateLimitException ] {
48
- TestFactory .mockedService429().createMessage(irrelevantMessages, None , settings)
50
+ TestFactory .mockedService429().createMessage(Some (role), irrelevantMessages , settings)
49
51
}
50
52
}
51
53
0 commit comments