fix: add Pinecone tests (#61)

gentrace · Apr 24, 2023 · 5828612 · 5828612
1 parent 5ad0814
commit 5828612
Show file tree

Hide file tree

Showing 7 changed files with 2,248 additions and 36 deletions.
diff --git a/package/conftest.py b/package/conftest.py
@@ -8,10 +8,11 @@
 from fixtures.completion import completion_response
 from fixtures.embedding import embedding_response
 from fixtures.gentrace import gentrace_pipeline_run_response
+from fixtures.vector import vector
 
 
 @pytest.fixture()
-def setupTeardown():
+def setupTeardownOpenAI():
     gentrace.api_key = os.getenv("GENTRACE_API_KEY")
     gentrace.host = "http://localhost:3000/api/v1"
 

diff --git a/package/fixtures/__init__.py b/package/fixtures/__init__.py
@@ -2,3 +2,4 @@
 from fixtures.completion import completion_response
 from fixtures.embedding import embedding_response
 from fixtures.gentrace import gentrace_pipeline_run_response
+from fixtures.vector import vector