fix: update paths in ollama tests

timescale · Oct 9, 2024 · f7884d3 · f7884d3
1 parent 1007fcd
commit f7884d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/projects/extension/tests/test_ollama.py b/projects/extension/tests/test_ollama.py
@@ -139,7 +139,7 @@ def test_ollama_image(cur_with_ollama_host):
         select ai.ollama_generate
         ( 'llava:7b'
         , 'Please describe this image.'
-        , _images=> array[pg_read_binary_file('/pgai/tests/postgresql-vs-pinecone.jpg')]
+        , _images=> array[pg_read_binary_file('/pgai/projects/extension/tests/postgresql-vs-pinecone.jpg')]
         , _system=>'you are a helpful assistant'
         , _options=> jsonb_build_object
           ( 'seed', 42
@@ -209,7 +209,7 @@ def test_ollama_chat_complete_image(cur_with_ollama_host):
           ( jsonb_build_object
             ( 'role', 'user'
             , 'content', 'describe this image'
-            , 'images', jsonb_build_array(encode(pg_read_binary_file('/pgai/tests/postgresql-vs-pinecone.jpg'), 'base64'))
+            , 'images', jsonb_build_array(encode(pg_read_binary_file('/pgai/projects/extension/tests/postgresql-vs-pinecone.jpg'), 'base64'))
             )
           )
         , _options=> jsonb_build_object