fix failing tests

svilupp · Nov 7, 2023 · fb441c9 · fb441c9
1 parent 30b68a0
commit fb441c9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/docs/make.jl b/docs/make.jl
@@ -13,6 +13,7 @@ makedocs(;
     sitename = "PromptingTools.jl",
     format = Documenter.HTML(;
         prettyurls = get(ENV, "CI", "false") == "true",
+        repolink = "https://github.com/svilupp/PromptingTools.jl",
         canonical = "https://svilupp.github.io/PromptingTools.jl",
         edit_link = "main",
         assets = String[]),

diff --git a/test/llm_openai.jl b/test/llm_openai.jl
@@ -1,4 +1,5 @@
 using PromptingTools: TestEchoOpenAISchema, render, OpenAISchema
+using PromptingTools: AIMessage, SystemMessage, UserMessage, DataMessage
 
 @testset "render-OpenAI" begin
     schema = OpenAISchema()

diff --git a/test/messages.jl b/test/messages.jl
@@ -1,3 +1,5 @@
+using PromptingTools: AIMessage, SystemMessage, UserMessage, DataMessage
+
 @testset "Message constructors" begin
     # Creates an instance of MSG with the given content string.
     content = "Hello, world!"