mainsequence 6.0.32__tar.gz → 6.0.34__tar.gz

{mainsequence-6.0.32/mainsequence.egg-info → mainsequence-6.0.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 6.0.32
+Version: 6.0.34
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: LicenseRef-MainSequence-Proprietary

{mainsequence-6.0.32 → mainsequence-6.0.34}/agent_scaffold/skills/a2a_sdk_execution/SKILL.md

@@ -39,9 +39,16 @@ internally and sends a standard A2A message.
 3. Run:
 
 ```bash
-mainsequence agent search "<discoveryPrompt>" --limit 10 --json
+mainsequence agent search "<discoveryPrompt>" \
+  --environment-uid "<organizationEnvironmentUid>" \
+  --limit 10 \
+  --json
 ```
 
+Use the canonical Organization Environment UID associated with the target
+ProjectBranch. It scopes discovery only; it does not create, assign, or override an
+Agent's environment.
+
 4. Treat the CLI output as authoritative.
 5. Prefer the highest `combined_score` when present.
 6. If the user asked only which agents are available, summarize the candidates and stop.
@@ -270,7 +277,7 @@ handle for a different task or conversation.
 ## Deterministic Execution Path
 
 1. Build the discovery prompt.
-2. Run `mainsequence agent search "<discoveryPrompt>" --limit <n> --json`.
+2. Run `mainsequence agent search "<discoveryPrompt>" --environment-uid "<organizationEnvironmentUid>" --limit <n> --json`.
 3. Select the target agent.
 4. Create or resolve the target session with `mainsequence agent session get_or_create`.
 5. Use the returned session `uid`.

{mainsequence-6.0.32 → mainsequence-6.0.34}/mainsequence/cli/api.py

@@ -1166,17 +1166,20 @@ def delete_organization_team(
 
 def list_agents(
     *,
+    organization_project_environment_uid: str,
     timeout: int | None = None,
     filters: dict[str, Any] | None = None,
 ) -> list[dict[str, Any]]:
     """
-    List agents via SDK client model.
+    List agents in one Organization Environment via the SDK client model.
     """
     try:
+        read_params = dict(filters or {})
+        read_params["organization_project_environment_uid"] = organization_project_environment_uid
         payload = _run_sdk_model_operation(
             module_name="mainsequence.client.agent_runtime_models",
             class_name="Agent",
-            operation=lambda ClientAgent: ClientAgent.filter(timeout=timeout, **(filters or {})),
+            operation=lambda ClientAgent: ClientAgent.filter(timeout=timeout, **read_params),
         )
         return [_sdk_object_to_dict(item) for item in list(payload or [])]
     except Exception as e:
@@ -1212,11 +1215,12 @@ def get_agent(
 def semantic_search_agents(
     q: str,
     *,
+    organization_project_environment_uid: str,
     limit: int = 20,
     timeout: int | None = None,
 ) -> list[dict[str, Any]]:
     """
-    Search agents via SDK client `Agent.semantic_search()`.
+    Search agents in one Organization Environment via `Agent.semantic_search()`.
     """
     try:
         payload = _run_sdk_model_operation(
@@ -1224,6 +1228,7 @@ def semantic_search_agents(
             class_name="Agent",
             operation=lambda ClientAgent: ClientAgent.semantic_search(
                 q,
+                organization_project_environment_uid=organization_project_environment_uid,
                 limit=limit,
                 timeout=timeout,
             ),

{mainsequence-6.0.32 → mainsequence-6.0.34}/mainsequence/cli/cli.py

@@ -3548,6 +3548,7 @@ def _format_agent_session_details(
 
 
 def _agent_list_impl(
+    organization_project_environment_uid: str,
     timeout: int | None,
     filter_entries: list[str] | None,
     show_filters: bool,
@@ -3561,7 +3562,11 @@ def _agent_list_impl(
     _require_login()
 
     try:
-        agents = list_agents(timeout=timeout, filters=filters)
+        agents = list_agents(
+            organization_project_environment_uid=organization_project_environment_uid,
+            timeout=timeout,
+            filters=filters,
+        )
     except ApiError as e:
         error(f"Agents fetch failed: {e}")
         raise typer.Exit(1) from e
@@ -3632,6 +3637,7 @@ def _agent_detail_impl(
 def _agent_search_impl(
     *,
     q: str,
+    organization_project_environment_uid: str,
     limit: int,
     timeout: int | None,
 ) -> None:
@@ -3640,6 +3646,7 @@ def _agent_search_impl(
     try:
         results = semantic_search_agents(
             q,
+            organization_project_environment_uid=organization_project_environment_uid,
             limit=limit,
             timeout=timeout,
         )
@@ -4766,6 +4773,11 @@ def _print_data_node_storage_search_section(
 
 @agent.command("list")
 def agent_list_cmd(
+    organization_project_environment_uid: uuid.UUID = typer.Option(
+        ...,
+        "--environment-uid",
+        help="Organization Environment UID that scopes Agent discovery.",
+    ),
     filter_entries: list[str] | None = typer.Option(None, "--filter", help=LIST_FILTER_OPTION_HELP),
     show_filters: bool = typer.Option(
         False, "--show-filters", help="Show the filters supported by this list command and exit."
@@ -4776,6 +4788,7 @@ def agent_list_cmd(
     List agents visible to the authenticated user.
     """
     _agent_list_impl(
+        organization_project_environment_uid=str(organization_project_environment_uid),
         timeout=timeout,
         filter_entries=filter_entries,
         show_filters=show_filters,
@@ -4801,6 +4814,11 @@ def agent_detail_cmd(
 @agent.command("search")
 def agent_search_cmd(
     q: str = typer.Argument(..., help="Natural-language query to match against agents."),
+    organization_project_environment_uid: uuid.UUID = typer.Option(
+        ...,
+        "--environment-uid",
+        help="Organization Environment UID that scopes Agent discovery.",
+    ),
     limit: int = typer.Option(
         20, "--limit", min=1, max=100, help="Maximum number of ranked agent matches to return."
     ),
@@ -4820,6 +4838,7 @@ def agent_search_cmd(
     """
     _agent_search_impl(
         q=q,
+        organization_project_environment_uid=str(organization_project_environment_uid),
         limit=limit,
         timeout=timeout,
     )

{mainsequence-6.0.32 → mainsequence-6.0.34}/mainsequence/client/agent_runtime_models.py

@@ -56,6 +56,22 @@ class AgentSemanticSearchResult(BasePydanticModel):
         "",
         description="Short description returned by semantic search for the matched agent.",
     )
+    project_branch_uid: str = Field(
+        ...,
+        description="Public UID of the ProjectBranch that owns the matched Project Coding Agent.",
+    )
+    repository_branch: str = Field(
+        ...,
+        description="Repository branch of the matched Project Coding Agent.",
+    )
+    organization_project_environment_uid: str = Field(
+        ...,
+        description="Public UID of the Organization Environment that scoped the search result.",
+    )
+    organization_project_environment_name: str = Field(
+        ...,
+        description="Name of the Organization Environment that scoped the search result.",
+    )
     semantic_score: float = Field(
         ...,
         description="Vector similarity component of the semantic-search ranking.",
@@ -243,6 +259,15 @@ class Agent(ShareableObjectMixin, BaseObjectOrm, BasePydanticModel):
         "agent_type": "str",
         "search": "str",
     }
+    READ_QUERY_PARAMS: ClassVar[dict[str, str]] = {
+        "organization_project_environment_uid": "uid",
+    }
+    READ_QUERY_PARAM_DESCRIPTIONS: ClassVar[dict[str, str]] = {
+        "organization_project_environment_uid": (
+            "Required Organization Environment boundary for Agent discovery. "
+            "This scopes reads and does not assign an environment to an Agent."
+        ),
+    }
 
     uid: str | None = Field(None, description="Public UID of the agent resource.")
     name: str = Field(
@@ -307,12 +332,34 @@ class Agent(ShareableObjectMixin, BaseObjectOrm, BasePydanticModel):
             "ProjectBranch-scoped."
         ),
     )
+    repository_branch: str | None = Field(
+        ...,
+        description=(
+            "Read-only repository branch projected from the canonical ProjectBranch, "
+            "or null when the Agent is not ProjectBranch-scoped."
+        ),
+    )
+    organization_project_environment_uid: str | None = Field(
+        ...,
+        description=(
+            "Read-only public UID of the Organization Environment derived from the "
+            "canonical ProjectBranch, or null when the Agent is not environment-scoped."
+        ),
+    )
+    organization_project_environment_name: str | None = Field(
+        ...,
+        description=(
+            "Read-only name of the Organization Environment derived from the canonical "
+            "ProjectBranch, or null when the Agent is not environment-scoped."
+        ),
+    )
 
     @classmethod
     def semantic_search(
         cls,
         q: str,
         *,
+        organization_project_environment_uid: str,
         limit: int = 20,
         timeout=None,
     ) -> list[AgentSemanticSearchResult]:
@@ -321,7 +368,7 @@ class Agent(ShareableObjectMixin, BaseObjectOrm, BasePydanticModel):
             POST <object_url>/semantic-search/
 
         Server behavior:
-        - results stay scoped to the caller organization
+        - results stay scoped to the requested Organization Environment
         - returns ranked lightweight agent search rows, not full Agent records
         """
         q = (q or "").strip()
@@ -333,6 +380,10 @@ class Agent(ShareableObjectMixin, BaseObjectOrm, BasePydanticModel):
             raise ValueError("limit must be between 1 and 100")
 
         body: dict[str, Any] = {
+            "organization_project_environment_uid": cls._coerce_filter_uid(
+                organization_project_environment_uid,
+                field_name="organization_project_environment_uid",
+            ),
             "q": q,
             "limit": limit,
         }

{mainsequence-6.0.32 → mainsequence-6.0.34}/mainsequence/client/models_helpers.py

@@ -734,6 +734,26 @@ class JobRun(BaseObjectOrm, BasePydanticModel):
         description="Read-only helper field containing the associated job name.",
         examples=["daily-training-job"],
     )
+    project_uid: str | None = Field(
+        ...,
+        description="Read-only public UID of the associated project, or null for a run outside project scope.",
+        examples=["1d0530c0-65d1-4db0-856b-dc29d8260a09"],
+    )
+    project_name: str | None = Field(
+        ...,
+        description="Read-only name of the associated project, or null for a run outside project scope.",
+        examples=["market-data-service"],
+    )
+    project_branch_uid: str | None = Field(
+        ...,
+        description="Read-only public UID of the associated project branch, or null for an unscoped run.",
+        examples=["5a28020a-0f1b-47ee-aab8-334286234bea"],
+    )
+    project_branch_name: str | None = Field(
+        ...,
+        description="Read-only repository branch name associated with the run, or null for an unscoped run.",
+        examples=["main"],
+    )
 
     execution_start: datetime.datetime | None = Field(
         default=None,

{mainsequence-6.0.32 → mainsequence-6.0.34/mainsequence.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 6.0.32
+Version: 6.0.34
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: LicenseRef-MainSequence-Proprietary

{mainsequence-6.0.32 → mainsequence-6.0.34}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mainsequence"
-version = "6.0.32"
+version = "6.0.34"
 description = "Main Sequence SDK "
 readme = "README.md"
 requires-python = ">=3.13"

{mainsequence-6.0.32 → mainsequence-6.0.34}/tests/test_cli.py

@@ -508,6 +508,7 @@ def test_list_organization_teams_uses_client_model(cli_mod, monkeypatch):
 def test_list_agents_uses_client_model(cli_mod, monkeypatch):
     api_mod = importlib.import_module("mainsequence.cli.api")
     captured = {}
+    environment_uid = "22222222-2222-4222-8222-222222222222"
 
     class FakeAgent:
         def __init__(self, uid, name):
@@ -532,12 +533,19 @@ def test_list_agents_uses_client_model(cli_mod, monkeypatch):
 
     monkeypatch.setattr(api_mod, "_run_sdk_model_operation", _run_sdk_model_operation)
 
-    out = api_mod.list_agents(timeout=9, filters={"agent_type": "custom"})
+    out = api_mod.list_agents(
+        organization_project_environment_uid=environment_uid,
+        timeout=9,
+        filters={"agent_type": "custom"},
+    )
     assert captured == {
         "module_name": "mainsequence.client.agent_runtime_models",
         "class_name": "Agent",
         "timeout": 9,
-        "filters": {"agent_type": "custom"},
+        "filters": {
+            "agent_type": "custom",
+            "organization_project_environment_uid": environment_uid,
+        },
     }
     assert out == [
         {
@@ -590,6 +598,7 @@ def test_get_project_repository_uses_public_client_model(cli_mod, monkeypatch):
 def test_semantic_search_agents_uses_client_model(cli_mod, monkeypatch):
     api_mod = importlib.import_module("mainsequence.cli.api")
     captured = {}
+    environment_uid = "22222222-2222-4222-8222-222222222222"
 
     class FakeSearchResult:
         def __init__(self):
@@ -616,8 +625,18 @@ def test_semantic_search_agents_uses_client_model(cli_mod, monkeypatch):
 
         class _ClientAgent:
             @classmethod
-            def semantic_search(cls, q, *, limit=20, timeout=None):
+            def semantic_search(
+                cls,
+                q,
+                *,
+                organization_project_environment_uid,
+                limit=20,
+                timeout=None,
+            ):
                 captured["q"] = q
+                captured["organization_project_environment_uid"] = (
+                    organization_project_environment_uid
+                )
                 captured["limit"] = limit
                 captured["timeout"] = timeout
                 return [FakeSearchResult()]
@@ -628,6 +647,7 @@ def test_semantic_search_agents_uses_client_model(cli_mod, monkeypatch):
 
     out = api_mod.semantic_search_agents(
         "data research",
+        organization_project_environment_uid=environment_uid,
         limit=10,
         timeout=17,
     )
@@ -635,6 +655,7 @@ def test_semantic_search_agents_uses_client_model(cli_mod, monkeypatch):
         "module_name": "mainsequence.client.agent_runtime_models",
         "class_name": "Agent",
         "q": "data research",
+        "organization_project_environment_uid": environment_uid,
         "limit": 10,
         "timeout": 17,
     }
@@ -5633,6 +5654,9 @@ def test_list_project_job_runs_uses_client_model(cli_mod, monkeypatch):
 
 def test_get_project_job_run_logs_uses_client_model(cli_mod, monkeypatch):
     api_mod = importlib.import_module("mainsequence.cli.api")
+    real_job_run = importlib.import_module(
+        "mainsequence.client.models_helpers"
+    ).JobRun
     captured = {}
 
     monkeypatch.setattr(
@@ -5658,22 +5682,38 @@ def test_get_project_job_run_logs_uses_client_model(cli_mod, monkeypatch):
     class FakeBaseObjectOrm:
         ROOT_URL = "https://old.test/api/v1"
 
-    class FakeJobRun:
-        ROOT_URL = "https://old.test/api/v1/job-runs"
+    class CanonicalJobRun(real_job_run):
 
         @classmethod
         def get(cls, pk, timeout=None):
             captured["job_run_uid_arg"] = pk
-            return types.SimpleNamespace(
-                get_logs=lambda timeout=None: {
-                    "job_run_uid": pk,
+            return cls.model_validate(
+                {
+                    "uid": pk,
+                    "name": "daily-run-1",
+                    "unique_identifier": "jobrun_abc123",
+                    "job_uid": "7d0ab07c-d1c0-4b7f-9c69-3c1a41c0a4da",
+                    "job_name": "daily-prices",
+                    "project_uid": "1d0530c0-65d1-4db0-856b-dc29d8260a09",
+                    "project_name": "market-data-service",
+                    "project_branch_uid": "5a28020a-0f1b-47ee-aab8-334286234bea",
+                    "project_branch_name": "main",
                     "status": "RUNNING",
-                    "rows": ["first line"],
+                    "runtime_image_uid": "6cfdb152-923e-45b9-a150-c4541c68b0d1",
+                    "runtime_image_digest": "sha256:" + "b" * 64,
                 }
             )
 
+        def get_logs(self, *, timeout=None):
+            captured["get_logs_timeout"] = timeout
+            return {
+                "job_run_uid": self.uid,
+                "status": self.status,
+                "rows": ["first line"],
+            }
+
     fake_base.BaseObjectOrm = FakeBaseObjectOrm
-    fake_helpers.JobRun = FakeJobRun
+    fake_helpers.JobRun = CanonicalJobRun
     fake_client_pkg.utils = fake_utils
 
     monkeypatch.setitem(sys.modules, "mainsequence.client", fake_client_pkg)
@@ -5683,6 +5723,7 @@ def test_get_project_job_run_logs_uses_client_model(cli_mod, monkeypatch):
 
     out = api_mod.get_project_job_run_logs("4c1d77c8-8a42-42b8-a9c1-06be9a336e5d")
     assert captured["job_run_uid_arg"] == "4c1d77c8-8a42-42b8-a9c1-06be9a336e5d"
+    assert captured["get_logs_timeout"] is None
     assert captured["jwt"] == ("acc", "ref")
     assert out == {
         "job_run_uid": "4c1d77c8-8a42-42b8-a9c1-06be9a336e5d",
@@ -6249,11 +6290,12 @@ def test_project_project_resource_delete_fastapi_requires_confirmation(
 
 
 def test_agent_list(cli_mod, runner, monkeypatch):
+    environment_uid = "22222222-2222-4222-8222-222222222222"
     monkeypatch.setattr(cli_mod, "_require_login", lambda: {"username": "u"})
     monkeypatch.setattr(
         cli_mod,
         "list_agents",
-        lambda timeout=None, filters=None: [
+        lambda organization_project_environment_uid, timeout=None, filters=None: [
             {
                 "uid": "e0e75693-4110-464c-93e0-82c7fd9c9a23",
                 "name": "Research Copilot",
@@ -6267,7 +6309,10 @@ def test_agent_list(cli_mod, runner, monkeypatch):
         ],
     )
 
-    result = runner.invoke(cli_mod.app, ["agent", "list"])
+    result = runner.invoke(
+        cli_mod.app,
+        ["agent", "list", "--environment-uid", environment_uid],
+    )
     assert result.exit_code == 0
     assert "Agents" in result.output
     assert "UID" in result.output
@@ -6278,11 +6323,12 @@ def test_agent_list(cli_mod, runner, monkeypatch):
 
 
 def test_agent_list_json(cli_mod, runner, monkeypatch):
+    environment_uid = "22222222-2222-4222-8222-222222222222"
     monkeypatch.setattr(cli_mod, "_require_login", lambda: {"username": "u"})
     monkeypatch.setattr(
         cli_mod,
         "list_agents",
-        lambda timeout=None, filters=None: [
+        lambda organization_project_environment_uid, timeout=None, filters=None: [
             {
                 "uid": "e0e75693-4110-464c-93e0-82c7fd9c9a23",
                 "name": "Research Copilot",
@@ -6296,7 +6342,10 @@ def test_agent_list_json(cli_mod, runner, monkeypatch):
         ],
     )
 
-    result = runner.invoke(cli_mod.app, ["agent", "list", "--json"])
+    result = runner.invoke(
+        cli_mod.app,
+        ["agent", "list", "--environment-uid", environment_uid, "--json"],
+    )
     assert result.exit_code == 0
     payload = json.loads(result.output)
     assert payload[0]["uid"] == "e0e75693-4110-464c-93e0-82c7fd9c9a23"
@@ -6305,12 +6354,14 @@ def test_agent_list_json(cli_mod, runner, monkeypatch):
 
 def test_agent_search(cli_mod, runner, monkeypatch):
     captured = {}
+    environment_uid = "22222222-2222-4222-8222-222222222222"
     monkeypatch.setattr(cli_mod, "_require_login", lambda: {"username": "u"})
 
-    def _search(q, *, limit=20, timeout=None):
+    def _search(q, *, organization_project_environment_uid, limit=20, timeout=None):
         captured.update(
             {
                 "q": q,
+                "organization_project_environment_uid": organization_project_environment_uid,
                 "limit": limit,
                 "timeout": timeout,
             }
@@ -6334,6 +6385,8 @@ def test_agent_search(cli_mod, runner, monkeypatch):
             "agent",
             "search",
             "data research",
+            "--environment-uid",
+            environment_uid,
             "--limit",
             "10",
             "--timeout",
@@ -6343,6 +6396,7 @@ def test_agent_search(cli_mod, runner, monkeypatch):
     assert result.exit_code == 0
     assert captured == {
         "q": "data research",
+        "organization_project_environment_uid": environment_uid,
         "limit": 10,
         "timeout": 17,
     }
@@ -6356,6 +6410,7 @@ def test_agent_search(cli_mod, runner, monkeypatch):
 
 
 def test_agent_search_json(cli_mod, runner, monkeypatch):
+    environment_uid = "22222222-2222-4222-8222-222222222222"
     monkeypatch.setattr(cli_mod, "_require_login", lambda: {"username": "u"})
     monkeypatch.setattr(
         cli_mod,
@@ -6372,7 +6427,17 @@ def test_agent_search_json(cli_mod, runner, monkeypatch):
         ],
     )
 
-    result = runner.invoke(cli_mod.app, ["agent", "search", "data research", "--json"])
+    result = runner.invoke(
+        cli_mod.app,
+        [
+            "agent",
+            "search",
+            "data research",
+            "--environment-uid",
+            environment_uid,
+            "--json",
+        ],
+    )
     assert result.exit_code == 0
     payload = json.loads(result.output)
     assert payload[0]["uid"] == "e0e75693-4110-464c-93e0-82c7fd9c9a23"