projectdavid-common 0.20.0__tar.gz → 0.21.1__tar.gz

{projectdavid_common-0.20.0/src/projectdavid_common.egg-info → projectdavid_common-0.21.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: projectdavid_common
-Version: 0.20.0
+Version: 0.21.1
 Summary: Common shared custom packages
 Author-email: "Francis N." <francis.neequaye@projectdavid.co.uk>
 License: MIT

{projectdavid_common-0.20.0 → projectdavid_common-0.21.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "projectdavid_common"
-version = "0.20.0"
+version = "0.21.1"
 description = "Common shared custom packages"
 readme = "README.md"
 authors = [

{projectdavid_common-0.20.0 → projectdavid_common-0.21.1}/src/projectdavid_common/constants/ai_model_map.py

@@ -62,45 +62,6 @@ HYPERBOLIC_MODELS = {
     "hyperbolic/openai/gpt-oss-20b": "openai/gpt-oss-20b",
 }
 
-GOOGLE_GEMINI_MODELS = {
-    "google/gemini-1.0-pro-vision-latest": "gemini-1.0-pro-vision-latest",
-    "google/gemini-pro-vision": "gemini-pro-vision",
-    "google/gemini-1.5-pro-latest": "gemini-1.5-pro-latest",
-    "google/gemini-1.5-pro-001": "gemini-1.5-pro-001",
-    "google/gemini-1.5-pro-002": "gemini-1.5-pro-002",
-    "google/gemini-1.5-pro": "gemini-1.5-pro",
-    "google/gemini-1.5-flash-latest": "gemini-1.5-flash-latest",
-    "google/gemini-1.5-flash-001": "gemini-1.5-flash-001",
-    "google/gemini-1.5-flash-001-tuning": "gemini-1.5-flash-001-tuning",
-    "google/gemini-1.5-flash": "gemini-1.5-flash",
-    "google/gemini-1.5-flash-002": "gemini-1.5-flash-002",
-    "google/gemini-1.5-flash-8b": "gemini-1.5-flash-8b",
-    "google/gemini-1.5-flash-8b-001": "gemini-1.5-flash-8b-001",
-    "google/gemini-1.5-flash-8b-latest": "gemini-1.5-flash-8b-latest",
-    "google/gemini-1.5-flash-8b-exp-0827": "gemini-1.5-flash-8b-exp-0827",
-    "google/gemini-1.5-flash-8b-exp-0924": "gemini-1.5-flash-8b-exp-0924",
-    "google/gemini-2.5-pro-exp-03-25": "gemini-2.5-pro-exp-03-25",
-    "google/gemini-2.5-pro-preview-03-25": "gemini-2.5-pro-preview-03-25",
-    "google/gemini-2.0-flash-exp": "gemini-2.0-flash-exp",
-    "google/gemini-2.0-flash": "gemini-2.0-flash",
-    "google/gemini-2.0-flash-001": "gemini-2.0-flash-001",
-    "google/gemini-2.0-flash-exp-image-generation": "gemini-2.0-flash-exp-image-generation",
-    "google/gemini-2.0-flash-lite-001": "gemini-2.0-flash-lite-001",
-    "google/gemini-2.0-flash-lite": "gemini-2.0-flash-lite",
-    "google/gemini-2.0-flash-lite-preview-02-05": "gemini-2.0-flash-lite-preview-02-05",
-    "google/gemini-2.0-flash-lite-preview": "gemini-2.0-flash-lite-preview",
-    "google/gemini-2.0-pro-exp": "gemini-2.0-pro-exp",
-    "google/gemini-2.0-pro-exp-02-05": "gemini-2.0-pro-exp-02-05",
-    "google/gemini-exp-1206": "gemini-exp-1206",
-    "google/gemini-2.0-flash-thinking-exp-01-21": "gemini-2.0-flash-thinking-exp-01-21",
-    "google/gemini-2.0-flash-thinking-exp": "gemini-2.0-flash-thinking-exp",
-    "google/gemini-2.0-flash-thinking-exp-1219": "gemini-2.0-flash-thinking-exp-1219",
-    "google/learnlm-1.5-pro-experimental": "learnlm-1.5-pro-experimental",
-    "google/gemma-3-1b-it": "gemma-3-1b-it",
-    "google/gemma-3-4b-it": "gemma-3-4b-it",
-    "google/gemma-3-12b-it": "gemma-3-12b-it",
-    "google/gemma-3-27b-it": "gemma-3-27b-it",
-}
 
 # --- MASTER COMBINED MAP ---
 # This merges them all into one flat lookup for the Router
@@ -108,5 +69,4 @@ MODEL_MAP = {
     **DEEPSEEK_NATIVE_MODELS,
     **TOGETHER_AI_MODELS,
     **HYPERBOLIC_MODELS,
-    **GOOGLE_GEMINI_MODELS,
 }

{projectdavid_common-0.20.0 → projectdavid_common-0.21.1}/src/projectdavid_common/schemas/actions_schema.py

@@ -1,4 +1,3 @@
-# src/projectdavid_common/schemas/actions_schema.py
 from datetime import datetime
 from enum import Enum
 from typing import Any, Dict, List, Optional
@@ -17,6 +16,10 @@ class ActionBase(BaseModel):
     function_args: Optional[Dict[str, Any]] = None
     result: Optional[Dict[str, Any]] = None
 
+    # New Fields
+    tool_call_id: Optional[str] = None
+    turn_index: Optional[int] = 0
+
     model_config = ConfigDict(from_attributes=True)
 
 
@@ -38,6 +41,10 @@ class ActionCreate(BaseModel):
     expires_at: Optional[datetime] = None
     status: Optional[str] = "pending"
 
+    # Optional input during creation (passed from Orchestrator)
+    tool_call_id: Optional[str] = None
+    turn_index: Optional[int] = 0
+
     @field_validator("tool_name", mode="before")
     @classmethod
     def validate_tool_fields(cls, v: Optional[str]) -> Optional[str]:
@@ -50,8 +57,8 @@ class ActionCreate(BaseModel):
             "example": {
                 "tool_name": "example_tool_name",
                 "run_id": "example_run_id",
-                "function_args": {"arg1": "value1", "arg2": "value2"},
-                "expires_at": "2024-09-10T12:00:00Z",
+                "tool_call_id": "call_abc123",
+                "function_args": {"arg1": "value1"},
                 "status": "pending",
             }
         }
@@ -63,6 +70,11 @@ class ActionRead(BaseModel):
     run_id: Optional[str] = None
     tool_id: Optional[str] = None
     tool_name: Optional[str] = None
+
+    # New Fields
+    tool_call_id: Optional[str] = None
+    turn_index: Optional[int] = None
+
     triggered_at: Optional[str] = None
     expires_at: Optional[str] = None
     is_processed: Optional[bool] = None
@@ -74,21 +86,6 @@ class ActionRead(BaseModel):
     model_config = ConfigDict(
         extra="forbid",
         validate_assignment=True,
-        json_schema_extra={
-            "example": {
-                "id": "action_123456",
-                "run_id": "run_123456",
-                "tool_id": "tool_123456",
-                "tool_name": "code_interpreter",
-                "triggered_at": "2025-03-24T12:00:00Z",
-                "expires_at": "2025-03-24T12:05:00Z",
-                "is_processed": False,
-                "processed_at": "2025-03-24T12:01:00Z",
-                "status": "in_progress",
-                "function_args": {"param1": "value1"},
-                "result": {"output": "result data"},
-            }
-        },
     )
 
 

{projectdavid_common-0.20.0 → projectdavid_common-0.21.1}/src/projectdavid_common/schemas/messages_schema.py

@@ -1,4 +1,4 @@
-#! python
+#! src/projectdavid_common/schemas/messages_schema.py
 from enum import Enum
 from typing import Any, Dict, List, Optional
 
@@ -20,6 +20,10 @@ class MessageCreate(BaseModel):
     assistant_id: str
     role: str  # Using string instead of Enum to allow flexible validation
     tool_id: Optional[str] = None
+
+    # --- Agentic Tracking ---
+    tool_call_id: Optional[str] = None  # ID of the call this message responds to
+
     meta_data: Optional[Dict[str, Any]] = None
     is_last_chunk: bool = False
 
@@ -48,9 +52,15 @@ class MessageCreate(BaseModel):
 
 class ToolMessageCreate(BaseModel):
     content: str
+    tool_call_id: Optional[str] = None
 
     model_config = ConfigDict(
-        json_schema_extra={"example": {"content": "This is the content of the tool message."}}
+        json_schema_extra={
+            "example": {
+                "content": "This is the content of the tool message.",
+                "tool_call_id": "call_abc123",
+            }
+        }
     )
 
 
@@ -68,6 +78,10 @@ class MessageRead(BaseModel):
     role: str
     run_id: Optional[str]
     tool_id: Optional[str] = None
+
+    # --- Agentic Tracking ---
+    tool_call_id: Optional[str] = None
+
     status: Optional[str]
     thread_id: str
     sender_id: Optional[str] = None
@@ -80,6 +94,7 @@ class MessageUpdate(BaseModel):
     meta_data: Optional[Dict[str, Any]]
     status: Optional[str]
     role: Optional[str]
+    tool_call_id: Optional[str] = None
 
     @field_validator("role", mode="before")
     @classmethod
@@ -111,7 +126,6 @@ class MessagesList(BaseModel):
 
 
 class MessageDeleted(BaseModel):
-
     id: str
     object: str = "thread.message.deleted"
     deleted: bool = True

{projectdavid_common-0.20.0 → projectdavid_common-0.21.1/src/projectdavid_common.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: projectdavid_common
-Version: 0.20.0
+Version: 0.21.1
 Summary: Common shared custom packages
 Author-email: "Francis N." <francis.neequaye@projectdavid.co.uk>
 License: MIT