oneshot-python 0.12.1__tar.gz → 0.14.0__tar.gz

{oneshot_python-0.12.1 → oneshot_python-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oneshot-python
-Version: 0.12.1
+Version: 0.14.0
 Summary: Core Python SDK for the OneShot API — HTTP client with x402 payment handling
 License-Expression: MIT
 Requires-Python: >=3.10

{oneshot_python-0.12.1 → oneshot_python-0.14.0}/oneshot/client.py

@@ -828,6 +828,22 @@ class OneShotClient:
         """Enrich a person's profile from LinkedIn URL, email, or name. Async."""
         return await self.acall_tool("/v1/tools/enrich/profile", kwargs)
 
+    def company_search(self, **kwargs: Any) -> Any:
+        """Search for companies by name, domain, industry, location, size, etc. Blocking."""
+        return self.call_tool("/v1/tools/research/company", kwargs)
+
+    async def acompany_search(self, **kwargs: Any) -> Any:
+        """Search for companies by name, domain, industry, location, size, etc. Async."""
+        return await self.acall_tool("/v1/tools/research/company", kwargs)
+
+    def enrich_company(self, **kwargs: Any) -> Any:
+        """Enrich a company from domain, name, LinkedIn URL, or ticker. Blocking."""
+        return self.call_tool("/v1/tools/enrich/company", kwargs)
+
+    async def aenrich_company(self, **kwargs: Any) -> Any:
+        """Enrich a company from domain, name, LinkedIn URL, or ticker. Async."""
+        return await self.acall_tool("/v1/tools/enrich/company", kwargs)
+
     def find_email(self, company_domain: str, *, full_name: Optional[str] = None, first_name: Optional[str] = None, last_name: Optional[str] = None, **kwargs: Any) -> Any:
         """Find a person's email address. Blocking."""
         payload: dict[str, Any] = {"company_domain": company_domain, **kwargs}
@@ -948,32 +964,44 @@ class OneShotClient:
 
     def tag_receipt_value(
         self,
-        receipt_id: str,
-        value_tag: dict[str, Any],
+        receipt_id: Optional[str] = None,
+        value_tag: Optional[dict[str, Any]] = None,
+        *,
+        request_id: Optional[str] = None,
     ) -> dict:
         """Tag a receipt with a value for RoCS computation. Blocking.
 
+        Identify the receipt by ``receipt_id`` (``rcpt_…``) or, equivalently, by
+        the ``request_id`` returned from the originating tool call — the API
+        resolves a ``request_id`` via the receipt's ``job_id``, so you can tag
+        value without a prior receipts lookup.
+
         ``value_tag`` shape: ``{"type": ..., "amount": ..., "label": ...}``.
         ``type`` must be one of ``revenue``, ``lead``, ``conversion``,
         ``savings``, ``engagement``. Tags are stored as ``pending`` until a
         judge service confirms them against inbound signals.
         """
         return asyncio.get_event_loop().run_until_complete(
-            self.atag_receipt_value(receipt_id, value_tag)
+            self.atag_receipt_value(receipt_id, value_tag, request_id=request_id)
         )
 
     async def atag_receipt_value(
         self,
-        receipt_id: str,
-        value_tag: dict[str, Any],
+        receipt_id: Optional[str] = None,
+        value_tag: Optional[dict[str, Any]] = None,
+        *,
+        request_id: Optional[str] = None,
     ) -> dict:
         """Tag a receipt with a value for RoCS computation. Async."""
-        if not receipt_id:
-            raise ValidationError("receipt_id is required", "receipt_id")
+        target = receipt_id or request_id
+        if not target:
+            raise ValidationError(
+                "receipt_id or request_id is required", "receipt_id"
+            )
         if not isinstance(value_tag, dict) or not value_tag.get("type"):
             raise ValidationError("value_tag.type is required", "value_tag.type")
         return await self.acall_free_patch(
-            f"/v1/analytics/receipts/{receipt_id}/value",
+            f"/v1/analytics/receipts/{target}/value",
             value_tag,
         )
 

{oneshot_python-0.12.1 → oneshot_python-0.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "oneshot-python"
-version = "0.12.1"
+version = "0.14.0"
 description = "Core Python SDK for the OneShot API — HTTP client with x402 payment handling"
 readme = {text = "Core Python SDK for the OneShot API", content-type = "text/plain"}
 license = "MIT"

{oneshot_python-0.12.1 → oneshot_python-0.14.0}/tests/test_tag_receipt_value.py

@@ -70,18 +70,52 @@ async def test_accepts_minimal_valid_tag_with_just_type():
     mock.assert_awaited_once()
 
 
+# ── Resolving by request_id (== job_id) instead of receipt_id ────────────
+
+@pytest.mark.asyncio
+async def test_tags_by_request_id():
+    # A request_id from the original tool call resolves server-side via job_id.
+    c, mock = _client_with_mocked_patch()
+    tag = {"type": "lead", "amount": 1}
+    await c.atag_receipt_value(value_tag=tag, request_id="0cbb87ee-1f2a-4c3d-9e8b-7a6f5d4c3b2a")
+    mock.assert_awaited_once_with(
+        "/v1/analytics/receipts/0cbb87ee-1f2a-4c3d-9e8b-7a6f5d4c3b2a/value", tag
+    )
+
+
+@pytest.mark.asyncio
+async def test_receipt_id_preferred_over_request_id():
+    c, mock = _client_with_mocked_patch()
+    tag = {"type": "revenue", "amount": 2}
+    await c.atag_receipt_value("rcpt_01HX", tag, request_id="some-uuid")
+    mock.assert_awaited_once_with("/v1/analytics/receipts/rcpt_01HX/value", tag)
+
+
+@pytest.mark.asyncio
+async def test_raises_when_neither_id_provided():
+    c, _ = _client_with_mocked_patch()
+    with pytest.raises(ValidationError) as excinfo:
+        await c.atag_receipt_value(value_tag={"type": "revenue", "amount": 1})
+    assert excinfo.value.field == "receipt_id"
+
+
 def test_sync_wrapper_delegates_to_async(monkeypatch):
     # The blocking ``tag_receipt_value`` should call through to the async
     # version via the event loop, matching every other tool method.
     c = OneShotClient(TEST_PRIVATE_KEY)
     called: dict = {}
 
-    async def fake_async(receipt_id, value_tag):
+    async def fake_async(receipt_id=None, value_tag=None, *, request_id=None):
         called["receipt_id"] = receipt_id
         called["value_tag"] = value_tag
+        called["request_id"] = request_id
         return {"ok": True}
 
     c.atag_receipt_value = fake_async  # type: ignore[method-assign]
     result = c.tag_receipt_value("rcpt_01HX", {"type": "savings", "amount": 12})
     assert result == {"ok": True}
-    assert called == {"receipt_id": "rcpt_01HX", "value_tag": {"type": "savings", "amount": 12}}
+    assert called == {
+        "receipt_id": "rcpt_01HX",
+        "value_tag": {"type": "savings", "amount": 12},
+        "request_id": None,
+    }

{oneshot_python-0.12.1 → oneshot_python-0.14.0}/uv.lock

@@ -549,7 +549,7 @@ wheels = [
 
 [[package]]
 name = "oneshot-python"
-version = "0.9.0"
+version = "0.14.0"
 source = { editable = "." }
 dependencies = [
     { name = "eth-account" },