codemie-test-harness 0.1.136__py3-none-any.whl → 0.1.138__py3-none-any.whl

codemie_test_harness/tests/__init__.py

@@ -37,6 +37,7 @@ LANGFUSE_TRACES_ENABLED = (
 
 PROJECT = os.getenv("PROJECT_NAME", "codemie")
 TEST_USER = os.getenv("TEST_USER_FULL_NAME", "Test User")
+GITHUB_URL = os.getenv("GITHUB_URL", "https://github.com")
 
 autotest_entity_prefix = (
     f"{''.join(random.choice(string.ascii_lowercase) for _ in range(3))}_"

codemie_test_harness/tests/assistant/test_assistants.py

@@ -220,6 +220,8 @@ def test_create_assistant_and_prompt_with_file(
     prompt = (
         f"What is the content/text of the {file_name}. Show information from ALL pages. "
         "For PDF use 'Text' for query. If provided file is CSV then run python_repl_ast tool and show first 5 rows."
+        "For images explain what you see on it."
+        "For excel show data from the sheet with name 'Sheet1'"
     )
 
     assistant = assistant(system_prompt=system_prompt_for_chatting_with_files)

codemie_test_harness/tests/conftest.py

@@ -22,6 +22,7 @@ from codemie_sdk.models.integration import (
     Integration,
 )
 from codemie_sdk.models.workflow import WorkflowCreateRequest, WorkflowMode, Workflow
+
 from codemie_test_harness.tests import PROJECT, autotest_entity_prefix
 from codemie_test_harness.tests.test_data.google_datasource_test_data import (
     GOOGLE_DOC_URL,
@@ -34,11 +35,13 @@ from codemie_test_harness.tests.utils.constants import TESTS_PATH
 from codemie_test_harness.tests.utils.datasource_utils import DataSourceUtils
 from codemie_test_harness.tests.utils.gitbud_utils import GitBudUtils
 from codemie_test_harness.tests.utils.integration_utils import IntegrationUtils
+from codemie_test_harness.tests.utils.llm_utils import LLMUtils
 from codemie_test_harness.tests.utils.logger_util import setup_logger
 from codemie_test_harness.tests.utils.notification_utils import GmailUtils
 from codemie_test_harness.tests.utils.provider_utils import ProviderUtils
 from codemie_test_harness.tests.utils.search_utils import SearchUtils
 from codemie_test_harness.tests.utils.similarity_check import SimilarityCheck
+from codemie_test_harness.tests.utils.user_utils import UserUtils
 from codemie_test_harness.tests.utils.workflow_utils import WorkflowUtils
 from codemie_test_harness.tests.utils.yaml_utils import (
     WorkflowYamlModel,
@@ -190,6 +193,16 @@ def search_utils(client):
     return SearchUtils(client)
 
 
+@pytest.fixture(scope="session")
+def user_utils(client):
+    return UserUtils(client)
+
+
+@pytest.fixture(scope="session")
+def llm_utils(client):
+    return LLMUtils(client)
+
+
 @pytest.fixture(scope="function")
 def git_integration(integration_utils):
     integration = integration_utils.create_integration(
@@ -434,6 +447,7 @@ def assistant(default_llm, assistant_utils):
         context=None,
         mcp_server=None,
         project_name: str = None,
+        description: str = None,
         system_prompt="You are a helpful integration test assistant",
     ):
         nonlocal created_assistant
@@ -461,6 +475,7 @@ def assistant(default_llm, assistant_utils):
             mcp_servers=[mcp_server] if mcp_server else [],
             system_prompt=system_prompt,
             project_name=project_name,
+            description=description,
         )
         return created_assistant
 
@@ -479,6 +494,7 @@ def workflow(workflow_utils):
         assistant_model=None,
         state_model=None,
         project_name=None,
+        description=None,
     ):
         workflow_name = workflow_name if workflow_name else get_random_name()
 
@@ -494,6 +510,7 @@ def workflow(workflow_utils):
         created_workflow = workflow_utils.create_workflow(
             workflow_name=workflow_name,
             workflow_type=WorkflowMode.SEQUENTIAL,
+            description=description,
             shared=True,
             workflow_yaml=yaml_content,
             project_name=project_name,