From fb441c95ef9423763f13d8df11307dff79951561 Mon Sep 17 00:00:00 2001
From: J S <49557684+svilupp@users.noreply.github.com>
Date: Tue, 7 Nov 2023 16:39:05 -0600
Subject: [PATCH] fix failing tests

---
 docs/make.jl       | 1 +
 test/llm_openai.jl | 1 +
 test/messages.jl   | 2 ++
 3 files changed, 4 insertions(+)

diff --git a/docs/make.jl b/docs/make.jl
index 7307e9393..c2e3db45d 100644
--- 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
index b5234b475..135c14547 100644
--- 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
index 507ff0522..0068712c3 100644
--- 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!"