From 23235f4aaf688319a8caa91b73f9ded7bd85c4ab Mon Sep 17 00:00:00 2001 From: Vedantsahai18 Date: Tue, 24 Dec 2024 05:03:51 +0000 Subject: [PATCH] refactor: Lint agents-api (CI) --- agents-api/tests/fixtures.py | 2 +- agents-api/tests/test_docs_queries.py | 16 +++++++--------- 2 files changed, 8 insertions(+), 10 deletions(-) diff --git a/agents-api/tests/fixtures.py b/agents-api/tests/fixtures.py index 86ee8b815..417cab825 100644 --- a/agents-api/tests/fixtures.py +++ b/agents-api/tests/fixtures.py @@ -21,7 +21,6 @@ from agents_api.queries.developers.create_developer import create_developer from agents_api.queries.developers.get_developer import get_developer from agents_api.queries.docs.create_doc import create_doc -from agents_api.queries.tools.delete_tool import delete_tool # from agents_api.queries.executions.create_execution import create_execution # from agents_api.queries.executions.create_execution_transition import create_execution_transition @@ -30,6 +29,7 @@ from agents_api.queries.sessions.create_session import create_session from agents_api.queries.tasks.create_task import create_task from agents_api.queries.tools.create_tools import create_tools +from agents_api.queries.tools.delete_tool import delete_tool from agents_api.queries.users.create_user import create_user from agents_api.web import app diff --git a/agents-api/tests/test_docs_queries.py b/agents-api/tests/test_docs_queries.py index 4e2006310..7eacaf1dc 100644 --- a/agents-api/tests/test_docs_queries.py +++ b/agents-api/tests/test_docs_queries.py @@ -1,6 +1,7 @@ -from ward import skip, test import asyncio +from ward import skip, test + from agents_api.autogen.openapi_model import CreateDocRequest from agents_api.clients.pg import create_db_pool from agents_api.queries.docs.create_doc import create_doc @@ -10,13 +11,7 @@ from agents_api.queries.docs.search_docs_by_embedding import search_docs_by_embedding from agents_api.queries.docs.search_docs_by_text import search_docs_by_text from agents_api.queries.docs.search_docs_hybrid import search_docs_hybrid -from tests.fixtures import ( - pg_dsn, - test_agent, - test_developer, - test_doc, - test_user -) +from tests.fixtures import pg_dsn, test_agent, test_developer, test_doc, test_user EMBEDDING_SIZE: int = 1024 @@ -219,6 +214,7 @@ async def _(dsn=pg_dsn, developer=test_developer, agent=test_agent): ) assert not any(d.id == doc_agent.id for d in docs_list) + @skip("text search: test container not vectorizing") @test("query: search docs by text") async def _(dsn=pg_dsn, agent=test_agent, developer=test_developer): @@ -253,12 +249,13 @@ async def _(dsn=pg_dsn, agent=test_agent, developer=test_developer): ) print("\nSearch results:", result) - + # More specific assertions assert len(result) >= 1, "Should find at least one document" assert any(d.id == doc.id for d in result), f"Should find document {doc.id}" assert result[0].metadata == {"test": "test"}, "Metadata should match" + @skip("embedding search: test container not vectorizing") @test("query: search docs by embedding") async def _(dsn=pg_dsn, agent=test_agent, developer=test_developer): @@ -291,6 +288,7 @@ async def _(dsn=pg_dsn, agent=test_agent, developer=test_developer): assert len(result) >= 1 assert result[0].metadata is not None + @skip("hybrid search: test container not vectorizing") @test("query: search docs by hybrid") async def _(dsn=pg_dsn, agent=test_agent, developer=test_developer):