planar 0.10.0__py3-none-any.whl → 0.11.0__py3-none-any.whl

planar/app.py

@@ -28,7 +28,6 @@ from planar.routers import (
     create_workflow_router,
 )
 from planar.routers.agents_router import create_agent_router
-from planar.routers.dataset_router import create_dataset_router
 from planar.routers.entity_router import create_entities_router
 from planar.routers.object_config_router import create_object_config_router
 from planar.routers.rule import create_rule_router
@@ -130,14 +129,27 @@ class PlanarApp:
             create_human_task_routes(),
             prefix="/human-tasks",
         )
-        self.router_v1.include_router(
-            create_dataset_router(),
-            prefix="/datasets",
-        )
+
+        if self.config.data:
+            try:
+                from planar.routers.dataset_router import create_dataset_router
+
+                self.router_v1.include_router(
+                    create_dataset_router(),
+                    prefix="/datasets",
+                )
+            except ImportError:
+                logger.error(
+                    "Data dependencies not installed. Ensure you install the `data` optional dependency in your project (planar[data])"
+                )
+                raise
 
         self.router_v1.include_router(
             create_info_router(
-                title=title or "Planar API", description=description or "Planar API"
+                title=title or "Planar API",
+                description=description or "Planar API",
+                config=self.config,
+                registry=self._object_registry,
             ),
             prefix="",
         )

planar/routers/info.py

@@ -1,16 +1,34 @@
+import importlib.metadata
+from typing import Literal, TypedDict
+
 from fastapi import APIRouter, Depends
 from pydantic import BaseModel
-from sqlalchemy.ext.asyncio import AsyncSession
-from sqlmodel import col, distinct, func, select
+from sqlmodel import col, func, select
+from sqlmodel.ext.asyncio.session import AsyncSession
 
+from planar.config import PlanarConfig, get_environment
+from planar.data.config import DataConfig
+from planar.files.storage.config import StorageConfig
 from planar.human.models import HumanTask, HumanTaskStatus
 from planar.logging import get_logger
-from planar.object_config import ConfigurableObjectType, ObjectConfiguration
+from planar.object_registry import ObjectRegistry
 from planar.session import get_session
 from planar.workflows.models import Workflow, WorkflowStatus
 
 logger = get_logger(__name__)
 
+StorageInfo = Literal["s3", "localdir", "azure_blob"]
+
+
+class DatasetsInfo(BaseModel):
+    catalog: Literal["duckdb", "postgres", "sqlite"]
+    storage: StorageInfo
+
+
+class SystemFeatures(BaseModel):
+    storage: StorageInfo | None = None
+    datasets: DatasetsInfo | None = None
+
 
 class SystemInfo(BaseModel):
     """Combined application information and system statistics"""
@@ -19,6 +37,11 @@ class SystemInfo(BaseModel):
     title: str
     description: str
 
+    version: str
+    environment: str
+
+    features: SystemFeatures
+
     # System stats
     total_workflow_runs: int = 0
     completed_runs: int = 0
@@ -27,7 +50,18 @@ class SystemInfo(BaseModel):
     active_agents: int = 0
 
 
-async def get_system_stats(session: AsyncSession = Depends(get_session)) -> dict:
+class SystemStats(TypedDict):
+    total_workflow_runs: int
+    completed_runs: int
+    in_progress_runs: int
+    pending_human_tasks: int
+    active_agents: int
+
+
+async def get_system_stats(
+    registry: ObjectRegistry,
+    session: AsyncSession = Depends(get_session),
+) -> SystemStats:
     """
     Get system-wide statistics directly from the database.
 
@@ -35,8 +69,10 @@ async def get_system_stats(session: AsyncSession = Depends(get_session)) -> dict
     rather than fetching all records and calculating in the application.
     """
     try:
+        agent_count = len(registry.get_agents())
+
         # Get workflow run counts
-        workflow_stats = await session.execute(
+        workflow_stats = await session.exec(
             select(
                 func.count().label("total_runs"),
                 func.count(col(Workflow.id))
@@ -47,42 +83,21 @@ async def get_system_stats(session: AsyncSession = Depends(get_session)) -> dict
                 .label("in_progress_runs"),
             ).select_from(Workflow)
         )
-        workflow_row = workflow_stats.one()
+        total_runs, completed_runs, in_progress_runs = workflow_stats.one()
 
         # Get pending human task count
-        human_task_query = await session.execute(
+        human_task_query = await session.exec(
             select(func.count())
             .select_from(HumanTask)
             .where(HumanTask.status == HumanTaskStatus.PENDING)
         )
-        pending_tasks = human_task_query.scalar() or 0
-
-        # Get agent count from the registry or count distinct agent configs
-        agent_count = 0
-        try:
-            # Count distinct agent names in the AgentConfig table
-            agent_query = await session.execute(
-                select(
-                    func.count(distinct(ObjectConfiguration.object_name))
-                ).select_from(
-                    select(ObjectConfiguration)
-                    .where(
-                        ObjectConfiguration.object_type == ConfigurableObjectType.AGENT
-                    )
-                    .subquery()
-                )
-            )
-            agent_count = agent_query.scalar() or 0
-        except Exception:
-            logger.exception("error counting agents")
-            # Fallback to 0
-            agent_count = 0
+        pending_tasks = human_task_query.one()
 
         # Return stats dict
         return {
-            "total_workflow_runs": workflow_row.total_runs or 0,
-            "completed_runs": workflow_row.completed_runs or 0,
-            "in_progress_runs": workflow_row.in_progress_runs or 0,
+            "total_workflow_runs": total_runs,
+            "completed_runs": completed_runs,
+            "in_progress_runs": in_progress_runs,
             "pending_human_tasks": pending_tasks,
             "active_agents": agent_count,
         }
@@ -98,13 +113,31 @@ async def get_system_stats(session: AsyncSession = Depends(get_session)) -> dict
         }
 
 
-def create_info_router(title: str, description: str) -> APIRouter:
+def get_app_version() -> str:
+    try:
+        return importlib.metadata.version("planar")
+    except importlib.metadata.PackageNotFoundError:
+        logger.warning("Planar package not found, returning development version")
+        return "development"
+
+
+def get_storage_info(cfg: StorageConfig) -> StorageInfo:
+    return cfg.backend
+
+
+def get_datasets_info(cfg: DataConfig) -> DatasetsInfo | None:
+    return DatasetsInfo(catalog=cfg.catalog.type, storage=get_storage_info(cfg.storage))
+
+
+def create_info_router(
+    title: str, description: str, config: PlanarConfig, registry: ObjectRegistry
+) -> APIRouter:
     """
     Create a router for serving combined application information and system statistics.
 
     This router provides a single endpoint to retrieve the application's title,
     description, and system-wide statistics on workflow runs, human tasks,
-    and registered agents.
+    and registered agents, as well as the application's features and configuration.
 
     Args:
         title: The application title
@@ -125,7 +158,17 @@ def create_info_router(title: str, description: str) -> APIRouter:
         Returns:
             SystemInfo object containing app details and system stats
         """
-        stats = await get_system_stats(session)
-        return SystemInfo(title=title, description=description, **stats)
+        stats = await get_system_stats(registry, session)
+        return SystemInfo(
+            title=title,
+            description=description,
+            version=get_app_version(),
+            environment=get_environment(),
+            features=SystemFeatures(
+                storage=get_storage_info(config.storage) if config.storage else None,
+                datasets=get_datasets_info(config.data) if config.data else None,
+            ),
+            **stats,
+        )
 
     return router

planar/scaffold_templates/pyproject.toml.j2

@@ -3,7 +3,7 @@ name = "{{ name }}"
 version = "0.1.0"
 requires-python = ">=3.12"
 dependencies = [
-    "planar>=0.9.0",
+    "planar[data]>=0.10.0",
 ]
 
 [[tool.uv.index]]

planar/testing/fixtures.py

@@ -36,7 +36,7 @@ from pathlib import Path
 import pytest
 
 from planar.app import PlanarApp
-from planar.config import load_config
+from planar.config import load_config, load_environment_aware_config
 from planar.data.config import DataConfig, SQLiteCatalogConfig
 from planar.db import DatabaseManager, new_session
 from planar.files.storage.config import LocalDirectoryConfig
@@ -138,9 +138,12 @@ def data_config(tmp_path):
 @pytest.fixture(name="app_with_data")
 def app_with_data_fixture(data_config):
     """Create a PlanarApp with data configuration."""
-    app = PlanarApp()
-    # Add data config to the app's config
-    app.config.data = data_config
+    config = load_environment_aware_config()
+
+    config.data = data_config
+
+    app = PlanarApp(config=config)
+
     return app
 
 

{planar-0.10.0.dist-info → planar-0.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: planar
-Version: 0.10.0
+Version: 0.11.0
 Summary: Add your description here
 License-Expression: LicenseRef-Proprietary
 Requires-Dist: aiofiles>=24.1.0
@@ -296,6 +296,14 @@ To disable SQLite testing:
 PLANAR_TEST_SQLITE=0 uv run pytest
 ```
 
+### Test coverage
+
+We use [pytest-cov](https://pypi.org/project/pytest-cov/) to measure test coverage. To generate a simple coverage report use the following command:
+
+```bash
+uv run pytest --cov=planar
+```
+
 ### Pre-commit hooks
 
 We use [pre-commit](https://pre-commit.com/) to manage pre-commit hooks. To install the pre-commit hooks, run the following command:

{planar-0.10.0.dist-info → planar-0.11.0.dist-info}/RECORD

@@ -6,10 +6,8 @@ planar/ai/agent_utils.py,sha256=MYNerdAm2TPVbDSKAmBCUlGmR56NAc8seZmDAFOWvUA,4199
 planar/ai/models.py,sha256=bZd4MoBBJMqzXJqsmsbMdZtOaRrNeX438CHAqOvmpfw,4598
 planar/ai/pydantic_ai.py,sha256=FpD0pE7wWNYwmEUZ90D7_J8gbAoqKmWtrLr2fhAd7rg,23503
 planar/ai/state.py,sha256=6vQ8VMLxJYB75QXkm8AVPkdXHUMwwjKxBWH-hGIK9p0,278
-planar/ai/test_agent_serialization.py,sha256=zYLIxhYdFhOZzBrEBoQNyYLyNcNxWwaMTkjt_ARTkZk,8073
-planar/ai/test_agent_tool_step_display.py,sha256=7OeKLnimItAkAFcArufDfiL3rXnQ1GwEtrE58_557hM,2588
 planar/ai/utils.py,sha256=WVBW0TGaoKytC4bNd_a9lXrBf5QsDRut4GBcA53U2Ww,3116
-planar/app.py,sha256=G5MtKJ4QPrUbiN9ykWCavVo2Kq4cMawZAxiS3uDyFaI,18673
+planar/app.py,sha256=SKP7xinjQZfVKssKdfq0eK7VaJru99ReGJpn5KeTD3w,19108
 planar/cli.py,sha256=SIyQOY3MbNDuohNcXFRIrHJuGxFNNC8C_ihfCXIUvbE,9900
 planar/config.py,sha256=6J42G9rEVUiOyCAY3EwUTU3PPmWthGTnrHMzST9TMcc,17809
 planar/data/__init__.py,sha256=LwrWl925w1CN0aW645Wpj_kDp0B8j5SsPzjr9iyrcmI,285
@@ -17,7 +15,6 @@ planar/data/config.py,sha256=zp6ChI_2MUMbupEVQNY-BxzcdLvejXG33DCp0BujGVU,1209
 planar/data/connection.py,sha256=OGBVVapxDu1-vxAHjlEmSQiapBXBegetb4Nu0wg4n2Q,3846
 planar/data/dataset.py,sha256=FKvuM2-xI101mEf1eeOYfNjSpHk8XjFXnAktTNR8Nig,5800
 planar/data/exceptions.py,sha256=AlhGQ_TReyEzfPSlqoXCjoZ1206Ut7dS4lrukVfGHaw,358
-planar/data/test_dataset.py,sha256=_tYvzw4y2-8rlRELdJxItcG_0OCFlOv7tBC7QI1j21o,10196
 planar/data/utils.py,sha256=hhHKpYBvzLdzjNy_OElpAqIU-3XuHqUkFCiF5YWsShs,2556
 planar/db/__init__.py,sha256=SNgB6unQ1f1E9dB9O-KrsPsYM17KLsgOW1u0ajqs57I,318
 planar/db/alembic/env.py,sha256=4yX_aCqrXScVN3q7ISiMEDz40-5Dmociqh6d7VCUPwc,6051
@@ -35,29 +32,20 @@ planar/files/storage/config.py,sha256=jE9Dn6cG_a4x9pdaZkasOxjyWkK6hmplLrPjEsRXGL
 planar/files/storage/context.py,sha256=Qadlz4T-FQ1A0TdGsOfuNmM679ohayU7oaexUpT8AY0,361
 planar/files/storage/local_directory.py,sha256=1SsEfr0Ud9TvSQJneDl_M-D7AFYixLE9t-bVIiY3aSI,7395
 planar/files/storage/s3.py,sha256=1861rSw3kplXtugUWD7mdSD_EnPSHME1mGc82V69r5g,8234
-planar/files/storage/test_azure_blob.py,sha256=OFYpns6JyeCCBHCoLz56uUHR6tWWeSZldUant5llczI,14200
-planar/files/storage/test_local_directory.py,sha256=KtzRfjtZUew1U-KETtD2mb6ywwX6HmjzaaeixOP0Ebg,5751
-planar/files/storage/test_s3.py,sha256=QG-CH7fiaRmQRwffnqG2mLRrw9LIlR2-xRyHs6Wuspo,10565
-planar/files/test_files.py,sha256=nclsbLnbijCWQ-Aj8Yvo06hs72PygL1Wps7uk7716sc,8957
 planar/human/__init__.py,sha256=FwpV-FFssKKlvKSjWoI4gJB1XTMaNb1UNCSBxjAtIBw,147
 planar/human/human.py,sha256=-oRtN_8bCtSV7Sxku7yG4rof7T5pr4j18Cfm3u4Z3PM,14925
 planar/human/models.py,sha256=Cec1Y9NGGtuAl1ZhqNc9PWIq__BbiWVTh7IYKR4yl3w,2317
-planar/human/test_human.py,sha256=GlbTjCNeAAfcwVIPq1TRiH0ojSgAU0i3oMc2yQe0xVo,13226
 planar/logging/__init__.py,sha256=BW101gskQS3C2agKSDiJEy4L-j5z6BP1AwEfd2JMXHM,254
 planar/logging/attributes.py,sha256=SB-lC-Osa55zXkX2UAWm56vKXvHgbF7TmhGnSiBWejk,1529
 planar/logging/context.py,sha256=mGVpL-276VWpRUIGlkGwbn9gkATqFS7lcFn1aM0v-eQ,384
 planar/logging/formatter.py,sha256=NMDhpo7ezsfGO7cPwOWbClqODviEgSjqHCoqrCCIu2c,3560
 planar/logging/logger.py,sha256=vj3-z2ea0J1ppgDpJBsrueA3jg5kjk9ceIo-4tWarXk,2650
 planar/logging/otel.py,sha256=bD1wgKTon1ooq3JIWc_i0_d_ejxw8_a5YPFuizRa5ic,1734
-planar/logging/test_formatter.py,sha256=nzlZFUC1JD4Imcy1H7VaFhwjrHzRc_oQLXWZXiN7iGA,11695
 planar/modeling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 planar/modeling/field_helpers.py,sha256=9SOHTWPzjlaiq7RF88wjug3NvAywFurcHn651YL_SNY,1903
 planar/modeling/json_schema_generator.py,sha256=NDqPkWQA_I7ywQXCEQfj5ub9u5KAFEcSQpXVkrCluV4,2864
 planar/modeling/mixins/__init__.py,sha256=Lwg5eL4VFfv61FRBvH5OZqIyfrSogxQlYLUDnWnSorg,320
 planar/modeling/mixins/auditable.py,sha256=WP7aDWVn1j22ZffKzYRpu23JQJ4vvHCU1qxcbgChwhc,1619
-planar/modeling/mixins/test_auditable.py,sha256=RSYesWWBysFEWTD39-yDhww3wCe1OPN9Yt3ywGmx4d8,2912
-planar/modeling/mixins/test_timestamp.py,sha256=oLKPvr8oUdjJPJRif81nn4YV_uwbxql_ojjXdKI7j7E,4366
-planar/modeling/mixins/test_uuid_primary_key.py,sha256=t9ZoB0dS4jjJVrHic7EeEh_g3eZeYV3mr0ylv3Kr1Io,1575
 planar/modeling/mixins/timestamp.py,sha256=-eHndCWztDiOxfCI2UknmphGeoHMJVDJG1Lz4KtkQUA,1641
 planar/modeling/mixins/uuid_primary_key.py,sha256=O1BtuXk5GdsfpwTS6nGZm1GNi0pdXKQ6Kt2f5ZjiuMc,453
 planar/modeling/orm/__init__.py,sha256=QwPgToKEf_gCdjAjKKmgh0xLTHGsboK1kH1a1a0tSy0,339
@@ -77,55 +65,28 @@ planar/routers/entity_router.py,sha256=7Y1LDSqI_ovoOGr9DGylGM8BmRxF-WSPQSwITJHc6
 planar/routers/event.py,sha256=yvzzMQaKRoS2A0KSjQTyWcfmBzpt8xPNDfVW50XUSCw,2961
 planar/routers/files.py,sha256=udu6PeZ9AuOpNyJete21rWAVQyE0qnC7tnSyJ97AH4Y,5644
 planar/routers/human.py,sha256=m_CbH97_QGmzriui_xopLOq-2D1kR0WgvO92fMaFeBQ,5010
-planar/routers/info.py,sha256=HQa-mumw4zitG61V9isJlZ3cMr8pEwlB54Ct_LrpJDo,4473
+planar/routers/info.py,sha256=_7edvZjWbI65htI8nrQdHNxjMfSn2u-JHmxnpNWUAiU,5396
 planar/routers/models.py,sha256=zknkVs9-B4UJlMe9fl2EpXx7sdzfjQfwAbNoL1a0wmI,4694
 planar/routers/object_config_router.py,sha256=zA8-gGBQp1-Gm3uCC4WJ6nLicFwt4CsCqCYLFp1lRN8,4802
 planar/routers/rule.py,sha256=d6giUwYRKzxQFPeoWbe8Ylp2Cxd71_uK8yoS9NrOOBg,3563
-planar/routers/test_agents_router.py,sha256=jzRCLB21YcEfhaFUos_gfRp9WDdP38_cozTQkHbi9b4,6099
-planar/routers/test_dataset_router.py,sha256=8Dm6ZFlIXQgwWLG3y00mz5cMsZvmIw7yM2-w-Qe_sCQ,13058
-planar/routers/test_files_router.py,sha256=HfZF1zeJ9BD2jhE6s698Jo7sDpO55RJF5g1Ksup4jtM,1576
-planar/routers/test_object_config_router.py,sha256=JpzoNlNONgljmGJYtrXnhtGhKjUT3YQMhP89fnt7dn4,11406
-planar/routers/test_routes_security.py,sha256=lXHeYg_th4UaDWeANM-dzONF8p2bEtwXJYYUlftE9R8,5556
-planar/routers/test_rule_router.py,sha256=08fa4sc7RaXvQzPCQQ4LaftfXuQwoPEDzcS4lesPG2Q,17220
-planar/routers/test_workflow_router.py,sha256=h2XTJrlO3vAshd45abtMaE0kigWTNxXjfJiuGcjcA7s,17463
 planar/routers/workflow.py,sha256=I9wJLqcFf52eEREhwqMsYICBEVD7I07g7ILWzxR5x9Y,18151
 planar/rules/__init__.py,sha256=lF3F8Rdf2ottjiJu0IeBdqhg1bckLhOqZFI2t-8KItM,474
 planar/rules/decorator.py,sha256=nxT17n9uwfXMOlk5lliw_cRS7Y83gMI6CQdrf_pB5yk,6666
 planar/rules/models.py,sha256=vC38JLeGzmU87L8BX4AyVJLJHmRYjWRmoHQ6S6ZlhPg,10186
 planar/rules/rule_configuration.py,sha256=B2G6mPnfxA277nF-Gr-B_Uely-ZOhz2jAhiwQMZuY-k,6508
 planar/rules/runner.py,sha256=KIPrt_ri50qotvDLOY9xly40bNTWRh8GVT2kEJFFtFo,1714
-planar/rules/test_data/account_dormancy_management.json,sha256=9aMMELZrF5DTBluMKUXJptxwULEcva4GHEyaapIeerY,4776
-planar/rules/test_data/airline_loyalty_points_calculator.json,sha256=7S1koMe60yR3h2VQys34oLy5ynhsEQ5wadMLPHCRQZA,5689
-planar/rules/test_data/applicant_risk_assessment.json,sha256=rj-Q13NczdNt00x5wrvGLalw5IfdT1j-_RvpwCZa7Fc,9994
-planar/rules/test_data/booking_fraud_detection.json,sha256=HCAsUIIlaPL8Ph3jtUFVi2vTZxZW53-XggjDWEoXmi4,10027
-planar/rules/test_data/cellular_data_rollover_system.json,sha256=8GjkKqqzK7usSa5KAvUwkD2CzW5JtLGECjlD5gH-vHk,6317
-planar/rules/test_data/clinical_trial_eligibility_screener.json,sha256=VV0RzWdThBNlIffuyklqAcKtRq-aOLHpcXX28IR9qOU,11152
-planar/rules/test_data/customer_lifetime_value.json,sha256=ZYeISNOsgT7zLcddeNrrrHlrocDlDBmSfOlQqhu3PTE,3562
-planar/rules/test_data/import_duties_calculator.json,sha256=JsXuvkiGQF6MxrgrIhFC1UxDCtO9nQpxtKKXwOQ1VYs,9062
-planar/rules/test_data/insurance_prior_authorization.json,sha256=27_jET4mmpuzLeOzVF5xi7c9IexX1UJTKsIoQGQ0n8c,12145
-planar/rules/test_data/online_check_in_eligibility_system.json,sha256=c6zCGojFN6c-CnuvG93zqlCZIrMccnOGAPMZVoMfirg,6114
-planar/rules/test_data/order_consolidation_system.json,sha256=kWJuVHAfAqsDW2xVdxHbtJTlfolQ_SM6_GISgnxsYB8,9803
-planar/rules/test_data/portfolio_risk_monitor.json,sha256=tTvQOJJLhakGxG4CnA9fdBIECstJnp0B8ogFADkdy8s,15168
-planar/rules/test_data/supply_chain_risk.json,sha256=fO0wV5ZnsZQpOP19Zp2troTMADaX0-KMpCxG_uHG198,7263
-planar/rules/test_data/warehouse_cross_docking.json,sha256=IPfcgNkY2sds301BeW6CjgFtK_zRyr27gI3UcqCB2Uo,5549
-planar/rules/test_rules.py,sha256=6M7CSg1bwn7O7DOoNi38vyVG4UmPQfRFxEO9qGE6rz0,52011
 planar/scaffold_templates/app/__init__.py.j2,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 planar/scaffold_templates/app/db/entities.py.j2,sha256=wg9O3JtRaRMKlDtoWHHodyNRL0s1UILvsr9fCQ_O2-4,279
 planar/scaffold_templates/app/flows/process_invoice.py.j2,sha256=R3EII_O2DHV1kvffW_AApZyaS6rR9eikcpxI08XH9dI,1691
 planar/scaffold_templates/main.py.j2,sha256=zrqsuv3Fp4lcknvB37RrRHy11msdFB1yDguYmTLLPhw,398
 planar/scaffold_templates/planar.dev.yaml.j2,sha256=I5-IqX7GJm6qA91WtUMw43L4hKACqgnER_H2racim4c,998
 planar/scaffold_templates/planar.prod.yaml.j2,sha256=FahJ2atDtvVH7IUCatGq6h9hmyF8meeiWC8RLfWphOQ,867
-planar/scaffold_templates/pyproject.toml.j2,sha256=nFfHWLp0sFK8cqjkdwBm6Hi6xsPzTNkaBeSgdTWTS-Q,183
+planar/scaffold_templates/pyproject.toml.j2,sha256=HpBc3cuTr-VeVN3BQ8y1Vr2eOtkL7TLH5mNz_T-akpA,190
 planar/security/auth_context.py,sha256=i63JkHQ3oXNlTis7GIKRkZJbkcvZhD2jVDuO7blgbSc,5068
 planar/security/auth_middleware.py,sha256=Grrm0i2bstWZ83ukrNZsHvFbNzffN0rvbbCcb2OxRY0,5746
 planar/security/authorization.py,sha256=zTh5rLmVJZnGDq540_1WYljJ7Hws-BY_P6VqUCONqLE,12760
 planar/security/default_policies.cedar,sha256=j7stjEfmgzthsDCjw1NwlE3_SxXQmLOGNRONM_2htoE,1492
 planar/security/security_context.py,sha256=vzfwiDJRzq8snaENrJddG5Ei-UUq_a5hkInMEOZf0mM,496
-planar/security/tests/test_auth_middleware.py,sha256=GjAyJYAR6jF_zGG85ZKY-E_e_aoqjK8qiHSalv2T2rk,6206
-planar/security/tests/test_authorization_context.py,sha256=cnsC3V13NBJwzyIwZaM9wu_vergcnmVhwB3khVH7G-o,2364
-planar/security/tests/test_cedar_basics.py,sha256=i1jLPjlJT1n_97onbeDYVpnwAzU2PmHvIPvaJSH1J2U,1026
-planar/security/tests/test_cedar_policies.py,sha256=-Vn_CQgCUAVg7YhdUd34FsOjNL1EmY_o92r-fzmknP8,4848
-planar/security/tests/test_jwt_principal_context.py,sha256=nGElTLtXbabkAxd3kXVpSFdH7kvSzHzSkp89g5Vu5Hc,4691
 planar/session.py,sha256=xLS9WPvaiy9nr2Olju1-C-7_sU5VXK8RuNdjuKndul4,1020
 planar/sse/constants.py,sha256=jE3SooTEWPuuL_Bi6DisJYMR9pKOiHVfboU2h5QTJRg,22
 planar/sse/example.html,sha256=SgTJbdJ3B1F1DxLC2YWuX2F1XVwKcTjX34CbJCXoCTM,4144
@@ -133,19 +94,11 @@ planar/sse/hub.py,sha256=5jhfk7zdCivau3TT1MxU2qtvETSskhqEiXzt-t0sRpE,6859
 planar/sse/model.py,sha256=fU_Fx9LS2ouS6-Dj1TIF-PLGul9YratKWafoWfZR1gc,123
 planar/sse/proxy.py,sha256=aJGo_-JIeQ0xSmE4HJdulZxIgCVRsBMMXqqSqtPvTvo,9177
 planar/task_local.py,sha256=pyvT0bdzAn15HL2yQUs9YrU5MVXh9njQt9MH51AGljs,1102
-planar/test_app.py,sha256=5dYhOW6lRbAx2X270DfqktkJ5IfuqfowX6bwxM1WQAM,4865
-planar/test_cli.py,sha256=faR6CSuooHHyyB5Yt-p8CIr7mGtKrrU2TLQbc4Oe9bA,13834
-planar/test_config.py,sha256=HcmDu1nwKZZhzHQLGVyP9oxje-_g_XubEsvzRj28QPg,14328
-planar/test_object_config.py,sha256=izn4s2HmSDWpGtgpOTDmKeUYN2-63WDR1QtVQrT-x00,20135
-planar/test_object_registry.py,sha256=R7IwbB2GACm2HUuVZTeVY4V12XB9_JgSSeppPxiCdfs,480
-planar/test_sqlalchemy.py,sha256=QTloaipWiFmlLTBGH6YCRkwi1R27gmQZnwprO7lPLfU,7058
-planar/test_utils.py,sha256=gKenXotj36SN_bb3bQpYPfD8t06IjnGBQqEgWpujHcA,3086
 planar/testing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-planar/testing/fixtures.py,sha256=2Pa5pdqrpcLNAxEsaZiDiLaivD4KWAL8pin22H-XuuM,9694
+planar/testing/fixtures.py,sha256=z0VkKTqAg9z_1L_ojB0pTzKSaLPcDeNxRq_RB36rUU8,9740
 planar/testing/memory_storage.py,sha256=apcuFisC3hW9KiU3kO8zwHQ6oK9Lu20NSX5fJ0LSZUY,2824
 planar/testing/planar_test_client.py,sha256=2atx5zER5bGrCVoa-lcag8xP2ZcbnOLKoGRKpfOJCg8,1992
 planar/testing/synchronizable_tracer.py,sha256=SWeta1CgwGsN5duC0FR8NyXOQ1b1L8nDpvGdjZVJ9Bg,4938
-planar/testing/test_memory_storage.py,sha256=So32XL0gbLDFMTl-WJN445x9jL6O8Qsqw8IRaiZnsPs,4797
 planar/testing/workflow_observer.py,sha256=0Q2xsYuZzNGXHZVwvXBqL9KXPsdIXuSZGBJAxHopzJw,2976
 planar/utils.py,sha256=YP37-ODS8nYOIfHPo11CwCpQRsg8oc57lQ0wkXwqCyo,3607
 planar/workflows/__init__.py,sha256=yFrrtKYUCx4jBPpHdEWDfKQgZXzGyr9voj5lFe9C-_w,826
@@ -166,14 +119,9 @@ planar/workflows/step_core.py,sha256=e-O-SP_1ufr3EYwuvu3O4I06aoqMxu_tpvHAUO-pMWg
 planar/workflows/step_metadata.py,sha256=7hwcIm6ot8m-iUXSYCbPmkg6bWegF6_RJ1stInvaFII,10903
 planar/workflows/step_testing_utils.py,sha256=WiTwxB4mM2y6dW7CJ3PlIR1BkBodSxQV7-S25pQ3Ycs,2361
 planar/workflows/sub_workflow_runner.py,sha256=EpS7DhhXRbC6ABm-Sho6Uyxh2TqCjcTPDYvcTQN4FjY,8313
-planar/workflows/test_concurrency_detection.py,sha256=yfgvLOMkPaK7EiW4ihm1KQx82Y-s9pB6uJhBfDi7PwQ,4528
-planar/workflows/test_lock_timeout.py,sha256=H78N090wJtiEg6SaJosfRWijpX6HwnyWyNNb7WaGPe0,5746
-planar/workflows/test_serialization.py,sha256=JfaveBRQTNMkucqkTorIMGcvi8S0j6uRtboFaWpCmes,39586
-planar/workflows/test_suspend_deserialization.py,sha256=ddw2jToSJ-ebQ0RfT7KWTRMCOs1nis1lprQiGIGuaJ0,7751
-planar/workflows/test_workflow.py,sha256=hBLPQYqUsWEQ_SopKgi69ckRC5OpmQEBlsPcftGMu_Q,65266
 planar/workflows/tracing.py,sha256=E7E_kj2VBQisDqrllviIshbvOmB9QcEeRwMapunqio4,2732
 planar/workflows/wrappers.py,sha256=dY_3NqkzGMG4jgX2lkAqvHTYFA1lBzhkQCw7N5CyaQM,1174
-planar-0.10.0.dist-info/WHEEL,sha256=-neZj6nU9KAMg2CnCY6T3w8J53nx1kFGw_9HfoSzM60,79
-planar-0.10.0.dist-info/entry_points.txt,sha256=L3T0w9u2UPKWXv6JbXFWKU1d5xyEAq1xVWbpYS6mLNg,96
-planar-0.10.0.dist-info/METADATA,sha256=yXTyA50EXqt_mZqhbAzm8Z2YRnrSmVzuCJLjvzf1qik,12348
-planar-0.10.0.dist-info/RECORD,,
+planar-0.11.0.dist-info/WHEEL,sha256=-neZj6nU9KAMg2CnCY6T3w8J53nx1kFGw_9HfoSzM60,79
+planar-0.11.0.dist-info/entry_points.txt,sha256=L3T0w9u2UPKWXv6JbXFWKU1d5xyEAq1xVWbpYS6mLNg,96
+planar-0.11.0.dist-info/METADATA,sha256=jlCvhwa-3Djk8_A05TJqgcGxE4Hj5U3wbYrDp5hNSi4,12556
+planar-0.11.0.dist-info/RECORD,,

planar/ai/test_agent_serialization.py

@@ -1,229 +0,0 @@
-"""
-Tests for agent serialization functionality.
-
-This module tests the serialization of agents including configuration
-management and schema validation warnings.
-"""
-
-import pytest
-from pydantic import BaseModel
-from sqlmodel.ext.asyncio.session import AsyncSession
-
-from planar.ai.agent import Agent
-from planar.ai.agent_utils import AgentConfig, agent_configuration
-from planar.ai.models import AgentSerializeable
-from planar.ai.utils import serialize_agent
-from planar.object_config.object_config import ObjectConfigurationBase
-
-
-class InputModelForTest(BaseModel):
-    """Test input model for agents."""
-
-    text: str
-    value: int
-
-
-class OutputModelForTest(BaseModel):
-    """Test output model for agents."""
-
-    result: str
-    score: float
-
-
-@pytest.fixture
-def test_agent():
-    """Create a test agent with various configurations."""
-    return Agent(
-        name="test_serialization_agent",
-        system_prompt="Test system prompt",
-        user_prompt="Test user prompt: {input}",
-        model="openai:gpt-4o",
-        max_turns=3,
-        input_type=InputModelForTest,
-        output_type=OutputModelForTest,
-    )
-
-
-@pytest.fixture
-def test_agent_with_tools():
-    """Create a test agent with tools."""
-
-    async def test_tool(param: str) -> str:
-        """A test tool."""
-        return f"Processed: {param}"
-
-    return Agent(
-        name="test_agent_with_tools",
-        system_prompt="System with tools",
-        user_prompt="User: {input}",
-        model="anthropic:claude-3-5-sonnet-latest",
-        max_turns=5,
-        tools=[test_tool],
-    )
-
-
-async def test_serialize_agent_basic(session: AsyncSession, test_agent):
-    """Test basic agent serialization without any configurations."""
-
-    # Serialize the agent
-    serialized = await serialize_agent(test_agent)
-
-    # Verify basic fields
-    assert isinstance(serialized, AgentSerializeable)
-    assert serialized.name == "test_serialization_agent"
-    assert serialized.input_schema is not None
-    assert serialized.output_schema is not None
-    assert serialized.tool_definitions == []
-
-    # Verify configs field exists and contains the default config (at least one config always present)
-    assert hasattr(serialized, "configs")
-    assert len(serialized.configs) == 1
-
-    # Verify the default config is present and correct
-    default_config = serialized.configs[-1]
-    assert isinstance(default_config, ObjectConfigurationBase)
-    assert default_config.version == 0
-    assert default_config.data.system_prompt == test_agent.system_prompt
-    assert default_config.data.user_prompt == test_agent.user_prompt
-    assert default_config.data.model == str(test_agent.model)
-    assert default_config.data.max_turns == test_agent.max_turns
-    assert default_config.data.model_parameters == test_agent.model_parameters
-
-    # Verify overwrites field is removed
-    assert not hasattr(serialized, "overwrites")
-
-
-async def test_serialize_agent_with_configs(session: AsyncSession, test_agent):
-    """Test agent serialization with multiple configurations."""
-
-    # Create multiple configurations
-    config1 = AgentConfig(
-        system_prompt="Override system 1",
-        user_prompt="Override user 1: {input}",
-        model="openai:gpt-4o",
-        max_turns=2,
-        model_parameters={"temperature": 0.7},
-    )
-
-    config2 = AgentConfig(
-        system_prompt="Override system 2",
-        user_prompt="Override user 2: {input}",
-        model="anthropic:claude-3-opus",
-        max_turns=4,
-        model_parameters={"temperature": 0.9},
-    )
-
-    # Write configurations
-    await agent_configuration.write_config(test_agent.name, config1)
-    await agent_configuration.write_config(test_agent.name, config2)
-
-    # Serialize the agent
-    serialized = await serialize_agent(test_agent)
-
-    # Verify configs are included
-    assert len(serialized.configs) == 3
-
-    # Verify default config is included
-    default_config = serialized.configs[-1]
-    assert isinstance(default_config, ObjectConfigurationBase)
-    assert default_config.version == 0
-    assert default_config.data.system_prompt == test_agent.system_prompt
-    assert default_config.data.user_prompt == test_agent.user_prompt
-    assert default_config.data.model == str(test_agent.model)
-    assert default_config.data.max_turns == test_agent.max_turns
-    assert default_config.data.model_parameters == test_agent.model_parameters
-
-    # Verify configs are ordered by version (descending)
-    assert all(
-        isinstance(config, ObjectConfigurationBase) for config in serialized.configs
-    )
-    assert serialized.configs[0].version == 2  # Latest version first
-    assert serialized.configs[1].version == 1
-
-    # Verify config data
-    latest_config = serialized.configs[0]
-    assert latest_config.data.system_prompt == "Override system 2"
-    assert latest_config.data.user_prompt == "Override user 2: {input}"
-    assert latest_config.data.model == "anthropic:claude-3-opus"
-    assert latest_config.data.max_turns == 4
-
-    older_config = serialized.configs[1]
-    assert older_config.data.system_prompt == "Override system 1"
-    assert older_config.data.user_prompt == "Override user 1: {input}"
-
-
-async def test_serialize_agent_with_tools(session: AsyncSession, test_agent_with_tools):
-    """Test serialization of agent with tools."""
-
-    # Serialize the agent
-    serialized = await serialize_agent(test_agent_with_tools)
-
-    # Verify tool definitions are included
-    assert len(serialized.tool_definitions) == 1
-    tool_def = serialized.tool_definitions[0]
-    assert tool_def["name"] == "test_tool"
-    assert tool_def["description"] == "A test tool."
-    assert "parameters" in tool_def
-
-
-async def test_serialize_agent_no_duplicate_fields(session: AsyncSession, test_agent):
-    """Test that AgentSerializeable doesn't duplicate fields from AgentConfig."""
-
-    # Create a configuration
-    config = AgentConfig(
-        system_prompt="Config system",
-        user_prompt="Config user: {input}",
-        model="openai:gpt-3.5-turbo",
-        max_turns=1,
-        model_parameters={},
-    )
-
-    await agent_configuration.write_config(test_agent.name, config)
-
-    # Serialize the agent
-    serialized = await serialize_agent(test_agent)
-
-    # Verify that system_prompt, user_prompt, model, max_turns are NOT in the serialized object
-    # They should only be in the configs
-    assert not hasattr(serialized, "system_prompt")
-    assert not hasattr(serialized, "user_prompt")
-    assert not hasattr(serialized, "model")
-    assert not hasattr(serialized, "max_turns")
-
-    # These fields should only be accessible through configs
-    assert serialized.configs[0].data.system_prompt == "Config system"
-    assert serialized.configs[0].data.user_prompt == "Config user: {input}"
-    assert serialized.configs[0].data.model == "openai:gpt-3.5-turbo"
-    assert serialized.configs[0].data.max_turns == 1
-
-
-async def test_agent_serializable_structure():
-    """Test the structure of AgentSerializeable model."""
-    # Verify the model has the expected fields
-    fields = AgentSerializeable.model_fields.keys()
-
-    # Should have these fields
-    assert "name" in fields
-    assert "input_schema" in fields
-    assert "output_schema" in fields
-    assert "tool_definitions" in fields
-    assert "configs" in fields
-    assert "built_in_vars" in fields
-
-    # Should NOT have these fields (moved to configs)
-    assert "system_prompt" not in fields
-    assert "user_prompt" not in fields
-    assert "model" not in fields
-    assert "max_turns" not in fields
-    assert "overwrites" not in fields
-
-
-async def test_configs_field_type():
-    """Test that configs field has the correct type annotation."""
-    # Get the type annotation for configs field
-    configs_field = AgentSerializeable.model_fields["configs"]
-
-    # The annotation should be list[ObjectConfigurationBase[AgentConfig]]
-    # This is a complex type, so we'll check the string representation
-    assert "ObjectConfigurationBase" in str(configs_field.annotation)
-    assert "AgentConfig" in str(configs_field.annotation)