everyrow 0.1.10__py3-none-any.whl → 0.2.0__py3-none-any.whl

everyrow/session.py

@@ -5,11 +5,11 @@ from datetime import datetime
 from uuid import UUID
 
 from everyrow.api_utils import create_client, handle_response
-from everyrow.generated.api.default import (
-    create_session_endpoint_sessions_create_post,
+from everyrow.generated.api.sessions import (
+    create_session_endpoint_sessions_post,
 )
 from everyrow.generated.client import AuthenticatedClient
-from everyrow.generated.models.create_session_request import CreateSessionRequest
+from everyrow.generated.models.create_session import CreateSession
 
 
 class Session:
@@ -61,11 +61,9 @@ async def create_session(
         await client.__aenter__()
 
     try:
-        response = await create_session_endpoint_sessions_create_post.asyncio(
+        response = await create_session_endpoint_sessions_post.asyncio(
             client=client,
-            body=CreateSessionRequest(
-                name=name or f"everyrow-sdk-session-{datetime.now().isoformat()}"
-            ),
+            body=CreateSession(name=name or f"everyrow-sdk-session-{datetime.now().isoformat()}"),
         )
         response = handle_response(response)
         session = Session(client=client, session_id=response.session_id)

everyrow/task.py

@@ -1,33 +1,36 @@
 import asyncio
-from typing import TypeVar, cast
+from enum import StrEnum
+from typing import TypeVar
 from uuid import UUID
 
 from pandas import DataFrame
 from pydantic.main import BaseModel
 
 from everyrow.api_utils import create_client, handle_response
-from everyrow.citations import render_citations_group, render_citations_standalone
 from everyrow.constants import EveryrowError
-from everyrow.generated.api.default import (
-    get_artifacts_artifacts_get,
-    get_task_status_endpoint_tasks_task_id_status_get,
-    submit_task_tasks_post,
+from everyrow.generated.api.tasks import (
+    get_task_result_tasks_task_id_result_get,
+    get_task_status_tasks_task_id_status_get,
 )
 from everyrow.generated.client import AuthenticatedClient
 from everyrow.generated.models import (
-    ArtifactGroupRecord,
-    LLMEnum,
-    StandaloneArtifactRecord,
-    TaskEffort,
+    LLMEnumPublic,
+    TaskResultResponse,
+    TaskResultResponseDataType1,
     TaskStatus,
     TaskStatusResponse,
 )
-from everyrow.generated.models.submit_task_body import SubmitTaskBody
+from everyrow.generated.types import Unset
 from everyrow.result import ScalarResult, TableResult
 
-# "export" generated types.
-LLM = LLMEnum
-EffortLevel = TaskEffort
+LLM = LLMEnumPublic
+
+
+class EffortLevel(StrEnum):
+    LOW = "low"
+    MEDIUM = "medium"
+    HIGH = "high"
+
 
 T = TypeVar("T", bound=BaseModel)
 
@@ -41,71 +44,57 @@ class EveryrowTask[T: BaseModel]:
         self._is_expand = is_expand
         self._response_model = response_model
 
-    async def submit(
+    def set_submitted(
         self,
-        body: SubmitTaskBody,
+        task_id: UUID,
+        session_id: UUID,
         client: AuthenticatedClient,
-    ) -> UUID:
-        task_id = await submit_task(body, client)
+    ) -> None:
         self.task_id = task_id
-        self.session_id = body.session_id
+        self.session_id = session_id
         self._client = client
-        return task_id
 
-    async def get_status(
-        self, client: AuthenticatedClient | None = None
-    ) -> TaskStatusResponse:
+    async def get_status(self, client: AuthenticatedClient | None = None) -> TaskStatusResponse:
         if self.task_id is None:
             raise EveryrowError("Task must be submitted before fetching status")
         client = client or self._client
         if client is None:
-            raise EveryrowError(
-                "No client available. Provide a client or use the task within a session context."
-            )
+            raise EveryrowError("No client available. Provide a client or use the task within a session context.")
         return await get_task_status(self.task_id, client)
 
-    async def await_result(
-        self, client: AuthenticatedClient | None = None
-    ) -> TableResult | ScalarResult[T]:
+    async def await_result(self, client: AuthenticatedClient | None = None) -> TableResult | ScalarResult[T]:
         if self.task_id is None:
             raise EveryrowError("Task must be submitted before awaiting result")
         client = client or self._client
         if client is None:
-            raise EveryrowError(
-                "No client available. Provide a client or use the task within a session context."
-            )
-        final_status_response = await await_task_completion(self.task_id, client)
-        artifact_id = cast(
-            UUID, final_status_response.artifact_id
-        )  # we check artifact_id in await_task_completion
+            raise EveryrowError("No client available. Provide a client or use the task within a session context.")
+        final_status = await await_task_completion(self.task_id, client)
+
+        result_response = await get_task_result(self.task_id, client)
+        artifact_id = result_response.artifact_id
+
+        if isinstance(artifact_id, Unset) or artifact_id is None:
+            raise EveryrowError("Task result has no artifact ID")
+
+        error = final_status.error if not isinstance(final_status.error, Unset) else None
 
         if self._is_map or self._is_expand:
-            data = await read_table_result(artifact_id, client=client)
+            data = _extract_table_data(result_response)
             return TableResult(
                 artifact_id=artifact_id,
                 data=data,
-                error=final_status_response.error,
+                error=error,
             )
         else:
-            data = await read_scalar_result(
-                artifact_id, self._response_model, client=client
-            )
+            data = _extract_scalar_data(result_response, self._response_model)
             return ScalarResult(
                 artifact_id=artifact_id,
                 data=data,
-                error=final_status_response.error,
+                error=error,
             )
 
 
-async def submit_task(body: SubmitTaskBody, client: AuthenticatedClient) -> UUID:
-    response = await submit_task_tasks_post.asyncio(client=client, body=body)
-    response = handle_response(response)
-    return response.task_id
-
-
-async def await_task_completion(
-    task_id: UUID, client: AuthenticatedClient
-) -> TaskStatusResponse:
+async def await_task_completion(task_id: UUID, client: AuthenticatedClient) -> TaskStatusResponse:
     max_retries = 3
     retries = 0
     while True:
@@ -113,9 +102,7 @@ async def await_task_completion(
             status_response = await get_task_status(task_id, client)
         except Exception as e:
             if retries >= max_retries:
-                raise EveryrowError(
-                    f"Failed to get task status after {max_retries} retries"
-                ) from e
+                raise EveryrowError(f"Failed to get task status after {max_retries} retries") from e
             retries += 1
         else:
             retries = 0
@@ -126,64 +113,42 @@ async def await_task_completion(
             ):
                 break
         await asyncio.sleep(1)
-    if (
-        status_response.status == TaskStatus.FAILED
-        or status_response.artifact_id is None
-    ):
-        raise EveryrowError(
-            f"Failed to create input in everyrow: {status_response.error}"
-        )
+
+    if status_response.status == TaskStatus.FAILED:
+        error_msg = status_response.error if not isinstance(status_response.error, Unset) else "Unknown error"
+        raise EveryrowError(f"Task failed: {error_msg}")
+
+    if status_response.status == TaskStatus.REVOKED:
+        raise EveryrowError("Task was revoked")
 
     return status_response
 
 
-async def get_task_status(
-    task_id: UUID, client: AuthenticatedClient
-) -> TaskStatusResponse:
-    response = await get_task_status_endpoint_tasks_task_id_status_get.asyncio(
-        client=client, task_id=task_id
-    )
+async def get_task_status(task_id: UUID, client: AuthenticatedClient) -> TaskStatusResponse:
+    response = await get_task_status_tasks_task_id_status_get.asyncio(task_id=task_id, client=client)
     response = handle_response(response)
     return response
 
 
-async def read_table_result(
-    artifact_id: UUID,
-    client: AuthenticatedClient,
-) -> DataFrame:
-    response = await get_artifacts_artifacts_get.asyncio(
-        client=client, artifact_ids=[artifact_id]
-    )
+async def get_task_result(task_id: UUID, client: AuthenticatedClient) -> TaskResultResponse:
+    response = await get_task_result_tasks_task_id_result_get.asyncio(task_id=task_id, client=client)
     response = handle_response(response)
-    if len(response) != 1:
-        raise EveryrowError(f"Expected 1 artifact, got {len(response)}")
-    artifact = response[0]
-    if not isinstance(artifact, ArtifactGroupRecord):
-        raise EveryrowError("Expected table result, but got a scalar")
-
-    artifact = render_citations_group(artifact)
-
-    return DataFrame([a.data for a in artifact.artifacts])
+    return response
 
 
-async def read_scalar_result[T: BaseModel](
-    artifact_id: UUID,
-    response_model: type[T],
-    client: AuthenticatedClient,
-) -> T:
-    response = await get_artifacts_artifacts_get.asyncio(
-        client=client, artifact_ids=[artifact_id]
-    )
-    response = handle_response(response)
-    if len(response) != 1:
-        raise EveryrowError(f"Expected 1 artifact, got {len(response)}")
-    artifact = response[0]
-    if not isinstance(artifact, StandaloneArtifactRecord):
-        raise EveryrowError("Expected scalar result, but got a table")
+def _extract_table_data(result: TaskResultResponse) -> DataFrame:
+    if isinstance(result.data, list):
+        records = [item.additional_properties for item in result.data]
+        return DataFrame(records)
+    raise EveryrowError("Expected table result (list of records), but got scalar or null")
 
-    artifact = render_citations_standalone(artifact)
 
-    return response_model(**artifact.data)
+def _extract_scalar_data[T: BaseModel](result: TaskResultResponse, response_model: type[T]) -> T:
+    if isinstance(result.data, TaskResultResponseDataType1):
+        return response_model(**result.data.additional_properties)
+    if isinstance(result.data, list) and len(result.data) == 1:
+        return response_model(**result.data[0].additional_properties)
+    raise EveryrowError("Expected scalar result, but got table or null")
 
 
 async def fetch_task_data(
@@ -192,9 +157,6 @@ async def fetch_task_data(
 ) -> DataFrame:
     """Fetch the result data for a completed task as a pandas DataFrame.
 
-    This is a convenience helper that retrieves the table-level group artifact
-    associated with a task and returns it as a DataFrame.
-
     Args:
         task_id: The UUID of the task to fetch data for (can be a string or UUID).
         client: Optional authenticated client. If not provided, one will be created
@@ -205,11 +167,6 @@ async def fetch_task_data(
 
     Raises:
         EveryrowError: If the task has not completed, failed, or has no artifact.
-
-    Example:
-        >>> from everyrow import fetch_task_data
-        >>> df = await fetch_task_data("12345678-1234-1234-1234-123456789abc")
-        >>> print(df.head())
     """
     if isinstance(task_id, str):
         task_id = UUID(task_id)
@@ -219,12 +176,8 @@ async def fetch_task_data(
 
     status_response = await get_task_status(task_id, client)
 
-    if status_response.status not in (TaskStatus.COMPLETED,):
-        raise EveryrowError(
-            f"Task {task_id} is not completed (status: {status_response.status.value}). Error: {status_response.error}"
-        )
-
-    if status_response.artifact_id is None:
-        raise EveryrowError(f"Task {task_id} has no associated artifact.")
+    if status_response.status != TaskStatus.COMPLETED:
+        raise EveryrowError(f"Task {task_id} is not completed (status: {status_response.status.value}).")
 
-    return await read_table_result(status_response.artifact_id, client)
+    result_response = await get_task_result(task_id, client)
+    return _extract_table_data(result_response)

{everyrow-0.1.10.dist-info → everyrow-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: everyrow
-Version: 0.1.10
+Version: 0.2.0
 Summary: An SDK for everyrow.io: agent ops at spreadsheet scale
 License-File: LICENSE.txt
 Requires-Python: >=3.12
@@ -213,7 +213,7 @@ result = await agent_map(
 print(result.data.head())
 ```
 
-**More:** [docs](docs/AGENT.md) / [basic usage](docs/case_studies/basic-usage/notebook.ipynb)
+**More:** [docs](docs/reference/RESEARCH.md) / [basic usage](docs/case_studies/basic-usage/notebook.ipynb)
 
 ### Derive
 
@@ -355,11 +355,12 @@ lefthook install
 ```
 
 ```bash
-uv run pytest              # tests
-uv run ruff check .        # lint
-uv run ruff format .       # format
-uv run basedpyright        # type check
-./generate_openapi.sh      # regenerate client
+uv run pytest                                          # unit tests
+uv run --env-file .env pytest -m integration           # integration tests (requires EVERYROW_API_KEY)
+uv run ruff check .                                    # lint
+uv run ruff format .                                   # format
+uv run basedpyright                                    # type check
+./generate_openapi.sh                                  # regenerate client
 ```
 
 ---
@@ -370,4 +371,17 @@ Built by [FutureSearch](https://futuresearch.ai). We kept running into the same
 
 [everyrow.io](https://everyrow.io) (app/dashboard) · [case studies](https://futuresearch.ai/solutions/) · [research](https://futuresearch.ai/research/)
 
-MIT license. See [LICENSE.txt](LICENSE.txt).
+**Citing everyrow:** If you use this software in your research, please cite it using the metadata in [CITATION.cff](CITATION.cff) or the BibTeX below:
+
+```bibtex
+@software{everyrow,
+  author       = {FutureSearch},
+  title        = {everyrow},
+  url          = {https://github.com/futuresearch/everyrow-sdk},
+  version      = {0.2.0},
+  year         = {2026},
+  license      = {MIT}
+}
+```
+
+**License** MIT license. See [LICENSE.txt](LICENSE.txt).

everyrow-0.2.0.dist-info/RECORD

@@ -0,0 +1,81 @@
+everyrow/__init__.py,sha256=z8oDcFaWns-sUIzAo063uzHRgPkptz1kAA2LhMTJ2Xw,328
+everyrow/api_utils.py,sha256=Wi3Yd30w11U02T_jt3GINebI2Ievhp8d42NHWxYAI4k,1641
+everyrow/billing.py,sha256=Oc0MsRl2CpnC8tH6srBnmdBNpgdygofKMKTdg8_MsLw,873
+everyrow/constants.py,sha256=oHWD2JAOeRySkf0K824XM6j07AzlTq_SUFvbpetrF1g,94
+everyrow/ops.py,sha256=rQUos96M97kurRHE3kHlvB2LDvYWaoJPj0khQ72BBkI,25906
+everyrow/result.py,sha256=2vCiE17kdbgkYKAdvfkpXJsSCr10U8FdO8NpS8eiofg,413
+everyrow/session.py,sha256=5ZB8bK3VtqC7VMKvKfyA1qKSYylOi-7XeXuSkLMsLRk,2656
+everyrow/task.py,sha256=UdGmt5OPQ4HGhx5iCRG6XlH2iUXZlkF7CV4y7mhDEqQ,6576
+everyrow/generated/__init__.py,sha256=7Zs5P68qSl1JSVmnLh4Uq-czi5SV3Fr7e923BVohTuA,155
+everyrow/generated/client.py,sha256=-rT3epMc77Y7QMTy5o1oH5hkGLufY9qFrD1rb7qItFU,12384
+everyrow/generated/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
+everyrow/generated/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
+everyrow/generated/types.py,sha256=0We4NPvhIYASRpQ3le41nmJeEAVm42-2VKdzlJ4Ogok,1343
+everyrow/generated/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
+everyrow/generated/api/artifacts/__init__.py,sha256=5vd9uJWAjRqa9xzxzYkLD1yoZ12Ld_bAaNB5WX4fbE8,56
+everyrow/generated/api/artifacts/create_artifact_artifacts_post.py,sha256=cs7-asfGObxTUtXPAhdeHkA-sqxx9oThAPfyqVYW4Rk,4847
+everyrow/generated/api/billing/__init__.py,sha256=5vd9uJWAjRqa9xzxzYkLD1yoZ12Ld_bAaNB5WX4fbE8,56
+everyrow/generated/api/billing/get_billing_balance_billing_get.py,sha256=3ddtkiRKQCEXPYg0Z451Yu8AKymhrl5rflWyMcDahSk,3429
+everyrow/generated/api/operations/__init__.py,sha256=5vd9uJWAjRqa9xzxzYkLD1yoZ12Ld_bAaNB5WX4fbE8,56
+everyrow/generated/api/operations/agent_map_operations_agent_map_post.py,sha256=GkW8ihrL5UtYlAmRZNSg85fuHZzql2ifvr0NlgNyb20,6081
+everyrow/generated/api/operations/dedupe_operations_dedupe_post.py,sha256=tUwWJsGBuFVN4wAgNvcnec5TVHVPMu8gqYP8rdBM5VM,5019
+everyrow/generated/api/operations/merge_operations_merge_post.py,sha256=aPZipbvuu_q_oPpPm6fNVbZ0S9NsfEQ9-JafZdwp32Y,4939
+everyrow/generated/api/operations/rank_operations_rank_post.py,sha256=xyJnMZOnkqKqZDxDr8ErVOrVCxVbe1ZbqPa9Edod3cE,4951
+everyrow/generated/api/operations/screen_operations_screen_post.py,sha256=riZ37utYJPBgrlVGPRGRczafIutUNSBy2dps0WAgYuk,5011
+everyrow/generated/api/operations/single_agent_operations_single_agent_post.py,sha256=yVIWTTxbkqQu1lDriYeHvo8qkbRneXPqZ78DlpYn2Tw,7761
+everyrow/generated/api/sessions/__init__.py,sha256=5vd9uJWAjRqa9xzxzYkLD1yoZ12Ld_bAaNB5WX4fbE8,56
+everyrow/generated/api/sessions/create_session_endpoint_sessions_post.py,sha256=CLjFKbfV2XD1tqJGTomqXXpMNctylzZ8I0rOL1MvOBA,4376
+everyrow/generated/api/tasks/__init__.py,sha256=5vd9uJWAjRqa9xzxzYkLD1yoZ12Ld_bAaNB5WX4fbE8,56
+everyrow/generated/api/tasks/get_task_result_tasks_task_id_result_get.py,sha256=mxRzES_LfgxTCiXdx50i2CXRHgvcDLbJzEIcYwSDGpo,5048
+everyrow/generated/api/tasks/get_task_status_tasks_task_id_status_get.py,sha256=W2FWklD65wEdu72fU5v8wzFb8McZmD32O5o3gzcfPsc,4672
+everyrow/generated/models/__init__.py,sha256=8AfDWIlIpCl2ub3xxXyYLr-KrBd9ZOCvRnMyZ0SwstU,4527
+everyrow/generated/models/agent_map_operation.py,sha256=K4vbipYRizEnUMVTeZcaE_b_81n8fln6RvH5007Av0Q,12671
+everyrow/generated/models/agent_map_operation_input_type_1_item.py,sha256=GgXSAQ4q4Tdw_aysRnHrCojHvmFi4w0Lnl65jLDkrlo,1351
+everyrow/generated/models/agent_map_operation_input_type_2.py,sha256=2Nl3D4Cnb3lSHGhHTMCnQUbcXPK5_aelAIi0LXhwois,1328
+everyrow/generated/models/agent_map_operation_response_schema_type_0.py,sha256=vOwu7fCzhtLJgm_brEvpRbFBp8A9Y0fCXnNBnBlh4bY,1376
+everyrow/generated/models/billing_response.py,sha256=wv8dTv6jZ5F5KjS3M5zwmDcf7r5ROYh5iTUbXWVwwMo,1743
+everyrow/generated/models/create_artifact_request.py,sha256=mZLGtjIILakvViSfFi-jWGWFhf7H2wQtG_KHzQbT4fo,4605
+everyrow/generated/models/create_artifact_request_data_type_0_item.py,sha256=vpQZVjBgmeDT4-799HBreVKeULghq24WKAikYVbSGNc,1366
+everyrow/generated/models/create_artifact_request_data_type_1.py,sha256=WwFB0AKrxqjoNntV3jaAoK4ZPXFnYGZlgxVityIlLlM,1343
+everyrow/generated/models/create_artifact_response.py,sha256=2JSlTpY3sS_DWCjgoBridc6QziBjPuTUrMsiyxkBIZs,1929
+everyrow/generated/models/create_session.py,sha256=wvqS0gYFo1q85Ow3E_uPnRavIjcYLKw1007nEzxnelg,1503
+everyrow/generated/models/dedupe_operation.py,sha256=i6rM5XAifdpvouyCk9MU6e35baragJwznDKULihibLo,5302
+everyrow/generated/models/dedupe_operation_input_type_1_item.py,sha256=nQocF04VEwIM2Mm0YKth7zSMwYAsfq2gWbv_95eJhPU,1338
+everyrow/generated/models/dedupe_operation_input_type_2.py,sha256=SxS0qQuU4nHT9yMW8f-Mu5MlgelfPG5HIrMb_aL6zgA,1315
+everyrow/generated/models/error_response.py,sha256=t1mtTt6dRzXQgit6slBktTzIycqKuWkxLvY6Ba__v54,3284
+everyrow/generated/models/error_response_details_type_0.py,sha256=mSLFizTmH-jTYn09eL5s3e1iI5nAdKWOpHnetKx-UGs,1315
+everyrow/generated/models/http_validation_error.py,sha256=ceDDkSqsvFmTdFwrra6lwo-L-BAHNX8h8bwaKyxPkEk,2317
+everyrow/generated/models/insufficient_balance_error.py,sha256=LqjyKHBPejWM_r1AsrcKmIPLDrWKyLbeHJBVY0nHmj0,2638
+everyrow/generated/models/llm_enum_public.py,sha256=AU2r8aadd-JkwhXVavmCZrRBecuakhNolIy9H8f2fAg,1350
+everyrow/generated/models/merge_operation.py,sha256=AqN5TShJQ7FgGeQKVhnNnTSU-besQurT2f9bL90X6T8,10890
+everyrow/generated/models/merge_operation_left_input_type_1_item.py,sha256=OuaH6nFuq1UQo8NXkmv2GeYLj_sWbvdLFVubnGt5Uhk,1356
+everyrow/generated/models/merge_operation_left_input_type_2.py,sha256=mjY5QnuREiIwV5qi79sWOxoIdSy6FjOHaT3fcOEIuk0,1333
+everyrow/generated/models/merge_operation_right_input_type_1_item.py,sha256=a_XH8yZB9HmTDWGfFAtkCEYCMHhBE5hXRhTwPO3F5rg,1361
+everyrow/generated/models/merge_operation_right_input_type_2.py,sha256=c24_J6xv7ASJP3LzPbML72J3NBFgHemRDYetdB-T4vs,1338
+everyrow/generated/models/merge_operation_use_web_search_type_0.py,sha256=C4pBVJssNaOkfimp86wgbTgKJuBllJ03EqVBTdgzEXk,184
+everyrow/generated/models/operation_response.py,sha256=7iuCVtyHir5AY7SPexeyY_kw_9YBkGVsH2wp_hbpnxk,3918
+everyrow/generated/models/public_effort_level.py,sha256=vPdbY13dO3q0KNKN8OTgXqqQf6v4526zquwlMt0g5V4,178
+everyrow/generated/models/public_task_type.py,sha256=2MCa8Z_zDWOEcUSr6ux08Sc7JEGGtxRbehWdKF2LsXU,221
+everyrow/generated/models/rank_operation.py,sha256=QmVqs83hQdCUzNZ_bBHiPlIfLQWawC3zBLq43j1MRdg,7397
+everyrow/generated/models/rank_operation_input_type_1_item.py,sha256=3jDUXFV87M9NCMHd64QQA5GzW68iQq9O0qnzaB8126I,1328
+everyrow/generated/models/rank_operation_input_type_2.py,sha256=ma5sofRgv9q9PvsBDIWXFGBwGPKICsk6FAYlsf12Dg4,1305
+everyrow/generated/models/rank_operation_response_schema_type_0.py,sha256=dUKIutYuompTBKqYe2znSPJBvfOnjlm2ns2nO_LrRVY,1353
+everyrow/generated/models/screen_operation.py,sha256=gHXaRqV2oHXh-7VJS46yvU1m16hMTOjkLfCowwYJ1bo,6918
+everyrow/generated/models/screen_operation_input_type_1_item.py,sha256=X2kSbeZXRBMjSUEeK0B5E8loRbfkkQiNTpi-HUx9QIw,1338
+everyrow/generated/models/screen_operation_input_type_2.py,sha256=26G_W1Zhx1MEOHoJP7Ty_X9HNVZKwqIAvpzzj-RZsqk,1315
+everyrow/generated/models/screen_operation_response_schema_type_0.py,sha256=3OMoKwmHLK81HgScTsNeL3TPZLajeWMfiRNng4r9DrE,1363
+everyrow/generated/models/session_response.py,sha256=1L27retPnHk_5FCgYpkDJ_8j5zpnIatsRVVkIsXl-5E,1618
+everyrow/generated/models/single_agent_operation.py,sha256=tkXU78MaAWmsySdzSQhMn04pii4G3iWBQf-aBAcw24U,11893
+everyrow/generated/models/single_agent_operation_input_type_1_item.py,sha256=Kze2hVfabLSsB47Vhy5Ck4R6SaPXw5PunB4CkTZNxAc,1366
+everyrow/generated/models/single_agent_operation_input_type_2.py,sha256=fKvpPQM-_N49jqJAdZUwwguv2DW-zWGZCj-hICcLQX0,1343
+everyrow/generated/models/single_agent_operation_response_schema_type_0.py,sha256=Z0gIRLrfZYmZqwDpqqt4iBk7u0_Dg2S-g-skshzPH04,1391
+everyrow/generated/models/task_result_response.py,sha256=3K22Ymj4jLe3YDbmnQHm0PNkzuJDu3iVJuMYMydW8fw,6280
+everyrow/generated/models/task_result_response_data_type_0_item.py,sha256=XTmW7RRPDX07Z4Oqg3JkRY8qrhFwizYtrwYXUAgIvMw,1351
+everyrow/generated/models/task_result_response_data_type_1.py,sha256=BT5NBWuIWtc6RNMqmRvcKrkcNIGo1eWYMsfGykkGuOk,1328
+everyrow/generated/models/task_status.py,sha256=wtFjPLXyNeLQrudS8IJgkQUU_gGx53OdliHlwJhaQPc,237
+everyrow/generated/models/task_status_response.py,sha256=6HDLnc-5MhUgXHisjoobUA6UI9VNortdAW2w1mvuV_s,6086
+everyrow/generated/models/validation_error.py,sha256=n8d_ZobQV26pm0KyDAKvIo93uOBhz2BH59jpJAKwoPY,2180
+everyrow-0.2.0.dist-info/METADATA,sha256=yUdzgHTprrteylODCwwpYqptj5xF-9a8s82GpOGShOE,13090
+everyrow-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+everyrow-0.2.0.dist-info/licenses/LICENSE.txt,sha256=8gN2nA06HyReyL7Mfu9nsBIpUF-B6wL5SJenlMRN8ac,1070
+everyrow-0.2.0.dist-info/RECORD,,

everyrow/citations.py

@@ -1,50 +0,0 @@
-from copy import deepcopy
-from typing import Any
-
-from everyrow.generated.models import (
-    ArtifactGroupRecord,
-    AuxDataSourceBank,
-    StandaloneArtifactRecord,
-)
-from everyrow.generated.types import Unset
-
-
-def _render_citations(
-    data: dict[str, Any], source_bank: AuxDataSourceBank
-) -> dict[str, Any]:
-    result = deepcopy(data)
-    for source_id, source_data in source_bank.to_dict().items():
-        for key, value in result.items():
-            if isinstance(value, str):
-                result[key] = value.replace(source_id, source_data["url"])
-            else:
-                result[key] = value
-    return result
-
-
-def render_citations_standalone(artifact: StandaloneArtifactRecord):
-    if isinstance(artifact.aux_data, Unset) or isinstance(
-        artifact.aux_data.source_bank, Unset
-    ):
-        return artifact
-    source_bank = (
-        artifact.aux_data.source_bank
-    )  # create reference simply to make the type checker happy before deepcopy.
-    artifact = deepcopy(artifact)
-    artifact.data = _render_citations(artifact.data, source_bank)
-    return artifact
-
-
-def render_citations_group(artifact: ArtifactGroupRecord) -> ArtifactGroupRecord:
-    artifact = deepcopy(artifact)
-    new_artifacts = []
-    for artifact_item in artifact.artifacts:
-        if isinstance(artifact_item, StandaloneArtifactRecord):
-            item_to_add = render_citations_standalone(artifact_item)
-        elif isinstance(artifact_item, ArtifactGroupRecord):
-            item_to_add = render_citations_group(artifact_item)
-        else:
-            item_to_add = artifact_item
-        new_artifacts.append(item_to_add)
-    artifact.artifacts = new_artifacts
-    return artifact