Moving weave_scorer.py to weave codebase

Author: neutralino1

weave/trace/weave_client.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import dataclasses
-import datetime
+from datetime import datetime, timezone
 import json
 import logging
 import os
@@ -480,9 +480,9 @@ class CallDict(TypedDict):
     summary: dict | None
     display_name: str | None
     attributes: dict | None
-    started_at: datetime.datetime | None
-    ended_at: datetime.datetime | None
-    deleted_at: datetime.datetime | None
+    started_at: datetime | None
+    ended_at: datetime | None
+    deleted_at: datetime | None
 
 
 @dataclasses.dataclass
@@ -500,9 +500,9 @@ class Call:
     summary: dict | None = None
     _display_name: str | Callable[[Call], str] | None = None
     attributes: dict | None = None
-    started_at: datetime.datetime | None = None
-    ended_at: datetime.datetime | None = None
-    deleted_at: datetime.datetime | None = None
+    started_at: datetime | None = None
+    ended_at: datetime | None = None
+    deleted_at: datetime | None = None
 
     # These are the live children during logging
     _children: list[Call] = dataclasses.field(default_factory=list)
@@ -1165,7 +1165,7 @@ def create_call(
         current_wb_run_id = safe_current_wb_run_id()
         check_wandb_run_matches(current_wb_run_id, self.entity, self.project)
 
-        started_at = datetime.datetime.now(tz=datetime.timezone.utc)
+        started_at = datetime.now(tz=timezone.utc)
         project_id = self._project_id()
 
         _should_print_call_link = should_print_call_link()
@@ -1233,7 +1233,7 @@ def finish_call(
 
         from weave.trace.api import _global_postprocess_output
 
-        ended_at = datetime.datetime.now(tz=datetime.timezone.utc)
+        ended_at = datetime.now(tz=timezone.utc)
         call.ended_at = ended_at
         original_output = output
 
@@ -1471,8 +1471,8 @@ def add_cost(
         llm_id: str,
         prompt_token_cost: float,
         completion_token_cost: float,
-        effective_date: datetime.datetime | None = datetime.datetime.now(
-            datetime.timezone.utc
+        effective_date: datetime | None = datetime.now(
+            timezone.utc
         ),
         prompt_token_cost_unit: str | None = "USD",
         completion_token_cost_unit: str | None = "USD",