Upload folder using huggingface_hub

client.py

@@ -6,17 +6,17 @@ from openenv.core.client_types import StepResult
 from openenv.core.env_server.types import State
 from openenv.core import EnvClient
 
-from .models import AdminAction, AdminObservation
+from .models import PranaAction, PranaObservation
 
 
-class PranaEnv(EnvClient[AdminAction, AdminObservation, State]):
+class PranaEnv(EnvClient[PranaAction, PranaObservation, State]):
     """
     Client for PRANA-Env.
 
     Example:
         >>> with PranaEnv(base_url="http://localhost:8000") as client:
         ...     client.reset()
-        ...     result = client.step(AdminAction(
+        ...     result = client.step(PranaAction(
         ...         action_type="query_db",
         ...         target="PatientDB",
         ...         field="hba1c",
@@ -25,12 +25,12 @@ class PranaEnv(EnvClient[AdminAction, AdminObservation, State]):
         ...     print(result.observation.query_result)  # "7.2"
     """
 
-    def _step_payload(self, action: AdminAction) -> Dict:
+    def _step_payload(self, action: PranaAction) -> Dict:
         return {k: v for k, v in action.model_dump().items() if v is not None}
 
-    def _parse_result(self, payload: Dict) -> StepResult[AdminObservation]:
+    def _parse_result(self, payload: Dict) -> StepResult[PranaObservation]:
         obs_data = payload.get("observation", {})
-        observation = AdminObservation(
+        observation = PranaObservation(
             query_result=obs_data.get("query_result", ""),
             active_task=obs_data.get("active_task", "t1"),
             policy_alerts=obs_data.get("policy_alerts", ""),

models.py

@@ -12,14 +12,14 @@ from pydantic import Field
 from openenv.core.env_server.types import Action, Observation
 
 
-class AdminAction(Action):
+class PranaAction(Action):
     """
     Action for PRANA-Env.
 
     For the smoke test, only action_type='query_db' is supported.
 
     Example:
-        >>> action = AdminAction(
+        >>> action = PranaAction(
         ...     action_type="query_db",
         ...     target="PatientDB",
         ...     field="hba1c",
@@ -55,7 +55,7 @@ class AdminAction(Action):
     )
 
 
-class AdminObservation(Observation):
+class PranaObservation(Observation):
     """
     Observation from PRANA-Env.
 

openenv_prana_env.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: openenv-prana_env
+Version: 0.1.0
+Summary: Prana Env environment for OpenEnv
+Requires-Python: >=3.10
+Requires-Dist: openenv-core[core]>=0.2.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

openenv_prana_env.egg-info/SOURCES.txt

@@ -0,0 +1,15 @@
+README.md
+pyproject.toml
+./__init__.py
+./client.py
+./models.py
+./test_client.py
+openenv_prana_env.egg-info/PKG-INFO
+openenv_prana_env.egg-info/SOURCES.txt
+openenv_prana_env.egg-info/dependency_links.txt
+openenv_prana_env.egg-info/entry_points.txt
+openenv_prana_env.egg-info/requires.txt
+openenv_prana_env.egg-info/top_level.txt
+server/__init__.py
+server/app.py
+server/prana_env_environment.py

openenv_prana_env.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

openenv_prana_env.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+server = prana_env.server.app:main

openenv_prana_env.egg-info/requires.txt

@@ -0,0 +1,5 @@
+openenv-core[core]>=0.2.0
+
+[dev]
+pytest>=8.0.0
+pytest-cov>=4.0.0

openenv_prana_env.egg-info/top_level.txt

@@ -0,0 +1 @@
+prana_env

server/app.py

@@ -36,15 +36,15 @@ except Exception as e:  # pragma: no cover
     ) from e
 
 # Import from local models.py (PYTHONPATH includes /app/env in Docker)
-from models import AdminAction, AdminObservation
+from models import PranaAction, PranaObservation
 from .prana_env_environment import PranaEnvironment
 
 
 # Create the app with web interface and README integration
 app = create_app(
     PranaEnvironment,
-    AdminAction,
-    AdminObservation,
+    PranaAction,
+    PranaObservation,
     env_name="prana_env",
     max_concurrent_envs=1,  # increase this number to allow more concurrent WebSocket sessions
 )

server/prana_env_environment.py

@@ -13,7 +13,7 @@ from uuid import uuid4
 from openenv.core.env_server.interfaces import Environment
 from openenv.core.env_server.types import State
 
-from models import AdminAction, AdminObservation
+from models import PranaAction, PranaObservation
 
 tag = "[prana_env/environment]"
 logger = logging.getLogger(__name__)
@@ -52,7 +52,7 @@ class PranaEnvironment(Environment):
         with open(path) as f:
             return json.load(f)
 
-    def reset(self, seed: int | None = None, episode_id: str | None = None, **kwargs) -> AdminObservation:
+    def reset(self, seed: int | None = None, episode_id: str | None = None, **kwargs) -> PranaObservation:
         patient_id: str | None = kwargs.get("patient_id")
         self._state = State(episode_id=episode_id or str(uuid4()), step_count=0)
         self._active_task = "t1"
@@ -60,14 +60,14 @@ class PranaEnvironment(Environment):
 
         logger.info(f"{tag} reset — episode={self._state.episode_id} patient_id={patient_id}")
 
-        return AdminObservation(
+        return PranaObservation(
             query_result="Episode reset. Ready for task t1: Initial Labs.",
             active_task=self._active_task,
             done=False,
             reward=0.0,
         )
 
-    def step(self, action: AdminAction) -> AdminObservation:  # type: ignore[override]
+    def step(self, action: PranaAction) -> PranaObservation:  # type: ignore[override]
         self._state.step_count += 1
         logger.info(
             f"{tag} step={self._state.step_count} action_type={action.action_type} "
@@ -78,14 +78,14 @@ class PranaEnvironment(Environment):
             return self._handle_query_db(action)
 
         logger.warning(f"{tag} Unsupported action_type={action.action_type}")
-        return AdminObservation(
+        return PranaObservation(
             query_result=f"NOT_SUPPORTED: action_type '{action.action_type}' not implemented yet.",
             active_task=self._active_task,
             done=False,
             reward=0.0,
         )
 
-    def _handle_query_db(self, action: AdminAction) -> AdminObservation:
+    def _handle_query_db(self, action: PranaAction) -> PranaObservation:
         db_name = (action.target or "").lower()
         field = (action.field or "").lower()
         patient_id = action.patient_id or self._patient_id
@@ -94,7 +94,7 @@ class PranaEnvironment(Environment):
 
         if db_name != "patientdb":
             logger.warning(f"{tag} Datastore '{db_name}' not available in Phase 1")
-            return AdminObservation(
+            return PranaObservation(
                 query_result=f"NOT_AVAILABLE: datastore '{action.target}' not loaded in Phase 1.",
                 active_task=self._active_task,
                 done=False,
@@ -102,7 +102,7 @@ class PranaEnvironment(Environment):
             )
 
         if not patient_id:
-            return AdminObservation(
+            return PranaObservation(
                 query_result="ERROR: patient_id is required for query_db.",
                 active_task=self._active_task,
                 done=False,
@@ -114,7 +114,7 @@ class PranaEnvironment(Environment):
 
         if not patient:
             logger.info(f"{tag} patient_id={patient_id} NOT_FOUND in PatientDB")
-            return AdminObservation(
+            return PranaObservation(
                 query_result=f"NOT_FOUND: patient '{patient_id}' not in PatientDB.",
                 active_task=self._active_task,
                 done=False,
@@ -123,7 +123,7 @@ class PranaEnvironment(Environment):
 
         if field not in patient:
             logger.info(f"{tag} field={field} NOT_FOUND for patient={patient_id}")
-            return AdminObservation(
+            return PranaObservation(
                 query_result=f"NOT_FOUND: field '{field}' not in PatientDB for patient '{patient_id}'.",
                 active_task=self._active_task,
                 done=False,
@@ -133,7 +133,7 @@ class PranaEnvironment(Environment):
         value = patient[field]
         if value is None:
             logger.info(f"{tag} field={field} is NULL for patient={patient_id}")
-            return AdminObservation(
+            return PranaObservation(
                 query_result=f"NOT_FOUND: field '{field}' has no recorded value for patient '{patient_id}'.",
                 active_task=self._active_task,
                 done=False,
@@ -141,7 +141,7 @@ class PranaEnvironment(Environment):
             )
 
         logger.info(f"{tag} query_db success field={field} value={value} patient={patient_id}")
-        return AdminObservation(
+        return PranaObservation(
             query_result=str(value),
             active_task=self._active_task,
             done=False,

test_client.py

@@ -0,0 +1,13 @@
+from prana_env.client import PranaEnv
+from prana_env.models import PranaAction
+
+with PranaEnv(base_url="http://localhost:8000") as client:
+	client.reset()
+	result = client.step(PranaAction(action_type="query_db",
+								    target="PatientDB",
+                                    field="hba1c",
+                                    patient_id="P001",
+    ))
+	print(result.observation.query_result)
+
+