gateforge-sdk 0.2.9__tar.gz → 0.2.11__tar.gz

{gateforge_sdk-0.2.9 → gateforge_sdk-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gateforge-sdk
-Version: 0.2.9
+Version: 0.2.11
 Summary: Privacy-first LLMOps SDK — Auto-init, decorators, session management, prompt system with cache
 Project-URL: Homepage, https://gateforge.dev
 Project-URL: Documentation, https://gateforge.dev/docs

{gateforge_sdk-0.2.9 → gateforge_sdk-0.2.11}/gateforge/client.py

@@ -189,6 +189,8 @@ class GatforgeClient:
         max_tokens: int | None = None,
         temperature: float | None = None,
         options: CallOptions | None = None,
+        messages: list | None = None,
+        system_instruction: str | None = None,
     ) -> tuple[str, float]:
         """
         Output guardrail evaluation + PII rehydration + cost calculation +
@@ -206,8 +208,13 @@ class GatforgeClient:
                 apply_on_fail(output_violations)
                 self._report_violations(output_violations, pii_context.get("thread_id"))
 
-        # PII rehydration
-        rehydrated = rehydrate_text(content, domain, backend, pii_context) if flags.pii_enabled else content
+        # PII rehydration — only if something was actually anonymized (non-empty entities)
+        # Calling rehydrate with an empty context dict raises "Missing context keys"
+        rehydrated = (
+            rehydrate_text(content, domain, backend, pii_context)
+            if flags.pii_enabled and pii_entities
+            else content
+        )
 
         # Cost
         cost_usd = calculate_cost(model, prompt_tokens, completion_tokens, provider) if flags.track_cost else 0.0
@@ -243,6 +250,34 @@ class GatforgeClient:
 
         # Agentic trace event (richer, waterfall-ready)
         if flags.tracing_enabled and conversation_id:
+            # Extract system prompt: OpenAI/Anthropic messages list or Gemini system_instruction
+            sys_prompt: str | None = None
+            if system_instruction:
+                sys_prompt = str(system_instruction)[:500]
+            elif messages:
+                for m in messages:
+                    if not isinstance(m, dict):
+                        continue
+                    if m.get("role") == "system":
+                        c = m.get("content", "")
+                        sys_prompt = (c if isinstance(c, str) else
+                                      next((p.get("text", "") for p in c if isinstance(p, dict) and p.get("type") == "text"), "")
+                                      )[:500]
+                        break
+
+            llm_meta: dict = {}
+            if temperature is not None:
+                llm_meta["temperature"] = temperature
+            if max_tokens is not None:
+                llm_meta["max_tokens"] = max_tokens
+            if sys_prompt:
+                llm_meta["system_prompt"] = sys_prompt
+            if messages:
+                # Count only non-system turns for a clean "context length" number
+                llm_meta["messages_count"] = sum(1 for m in messages if isinstance(m, dict) and m.get("role") != "system")
+            if finish_reason:
+                llm_meta["finish_reason"] = finish_reason
+
             emit_trace_event(
                 api_key=self.api_key,
                 base_url=self.base_url,
@@ -261,6 +296,7 @@ class GatforgeClient:
                 experiment_id=ab_assignment.experiment_id if ab_assignment else None,
                 variant=ab_assignment.variant if ab_assignment else None,
                 content_preview=rehydrated[:500] if rehydrated else None,
+                metadata=llm_meta or None,
             )
 
         return rehydrated, cost_usd
@@ -466,6 +502,7 @@ class GatforgeClient:
             finish_reason=finish_reason,
             max_tokens=kwargs.get("max_tokens"),
             temperature=kwargs.get("temperature"),
+            messages=anon_messages,
         )
 
         return GatforgeResponse(

{gateforge_sdk-0.2.9 → gateforge_sdk-0.2.11}/gateforge/context.py

@@ -227,7 +227,7 @@ def agent(
                 event_type="agent_start",
                 step=start_step,
                 content_preview=user_input[:500] if user_input else None,
-                metadata={"user_id": user_id, "args": _safe_str(args)[:200], "agent_name": name or fn.__name__, "nested": is_nested},
+                metadata={"agent_name": name or fn.__name__},
             )
 
             # Enter trace context (only if not already in one)
@@ -248,7 +248,6 @@ def agent(
                     event_type="agent_end",
                     step=active_trace.step if active_trace else ctx.ctx.step,
                     content_preview=result_preview,
-                    metadata={"success": True},
                 )
 
                 return result

{gateforge_sdk-0.2.9 → gateforge_sdk-0.2.11}/gateforge/wrappers/gemini.py

@@ -175,6 +175,8 @@ class _WrappedModels:
             completion_tokens=getattr(usage, "candidates_token_count", 0) or 0,
             latency_ms=latency_ms, pii_entities=pii_entities, pii_context=pii_ctx,
             ab_assignment=ab_assignment, options=options,
+            messages=anon_contents,
+            system_instruction=str(kwargs["system_instruction"])[:500] if kwargs.get("system_instruction") else None,
         )
         if content != rehydrated:
             _patch_text(response, rehydrated)

{gateforge_sdk-0.2.9 → gateforge_sdk-0.2.11}/gateforge/wrappers/openai.py

@@ -94,6 +94,7 @@ class _WrappedCompletions:
             ab_assignment=ab_assignment,
             finish_reason=finish_reason,
             max_tokens=kwargs.get("max_tokens"), temperature=kwargs.get("temperature"),
+            messages=anon_messages,
             options=options,
         )
         if response.choices[0].message.content != rehydrated:
@@ -225,6 +226,7 @@ class _WrappedAsyncCompletions:
             ab_assignment=ab_assignment,
             finish_reason=finish_reason,
             max_tokens=kwargs.get("max_tokens"), temperature=kwargs.get("temperature"),
+            messages=anon_messages,
             options=options,
         )
         if response.choices[0].message.content != rehydrated:

{gateforge_sdk-0.2.9 → gateforge_sdk-0.2.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "gateforge-sdk"
-version = "0.2.9"
+version = "0.2.11"
 description = "Privacy-first LLMOps SDK — Auto-init, decorators, session management, prompt system with cache"
 readme = "README.md"
 license = { text = "MIT" }