Fix embeddings api_key (#31)

Author: svilupp

CHANGELOG.md

@@ -13,7 +13,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [0.10.0]
 
 ### Updated
-- Increased compat for PromptingTools to v0.85
+- Moved up compat for PromptingTools to v0.87 (fixing api key bug for embedding models)
+
+### Fixed
+- Fixed mock embedding response in the tests
 
 ## [0.9.0]
 

Project.toml

@@ -29,7 +29,7 @@ HTTP = "1.10.15"
 JSON3 = "1.14.1"
 LinearAlgebra = "1"
 ProgressMeter = "1.10"
-PromptingTools = "0.84 - 0.85"
+PromptingTools = "0.87"
 Random = "1"
 Snowball = "0.1"
 SparseArrays = "1"

test/preparation.jl

@@ -65,7 +65,9 @@ end
     @test_throws AssertionError get_embeddings(BitPackedBatchEmbedder(), String[])
 
     # corresponds to OpenAI API v1
-    response1 = Dict(:data => [Dict(:embedding => ones(128, 2))],
+    response1 = Dict(
+        :data => [Dict(:embedding => ones(Float32, 128)),
+            Dict(:embedding => ones(Float32, 128))],
         :usage => Dict(:total_tokens => 2, :prompt_tokens => 2, :completion_tokens => 0))
     schema = TestEchoOpenAISchema(; response = response1, status = 200)
     PT.register_model!(; name = "mock-emb", schema)

test/runtests.jl

@@ -9,6 +9,13 @@ using Aqua
 const PT = PromptingTools
 const RT = RAGTools
 
+# Set a dummy API key for tests using CustomOpenAISchema with mock HTTP servers
+# Required since OpenAI.jl now throws an error when api_key is empty
+# We set the PT global directly since it's loaded at module load time from preferences/ENV
+if isempty(PT.OPENAI_API_KEY)
+    PT.OPENAI_API_KEY = "test-api-key-for-mock-server"
+end
+
 @testset "RAGTools.jl" begin
     @testset "Code quality (Aqua.jl)" begin
         Aqua.test_all(RAGTools)