Set logfire OTEL scope to magentic (#298)

* Set logfire OTEL scope to magentic

* Add TODOs for setting scope

* Set logfire otel_scope to magentic

* Add TODO for otel version

src/magentic/chat_model/retry_chat_model.py

@@ -2,8 +2,6 @@
 from functools import singledispatchmethod
 from typing import Any, TypeVar, overload
 
-import logfire_api as logfire
-
 from magentic.chat_model.base import (
     ChatModel,
     ToolSchemaParseError,
@@ -13,6 +11,7 @@
     Message,
     ToolResultMessage,
 )
+from magentic.logger import logfire
 
 R = TypeVar("R")
 

src/magentic/chatprompt.py

@@ -13,12 +13,11 @@
     overload,
 )
 
-import logfire_api as logfire
-
 from magentic.backend import get_chat_model
 from magentic.chat_model.base import ChatModel
 from magentic.chat_model.message import Message
 from magentic.chat_model.retry_chat_model import RetryChatModel
+from magentic.logger import logfire
 from magentic.typing import split_union_type
 
 P = ParamSpec("P")

src/magentic/function_call.py

@@ -16,8 +16,7 @@
 )
 from uuid import uuid4
 
-import logfire_api as logfire
-
+from magentic.logger import logfire
 from magentic.streaming import CachedAsyncIterable, CachedIterable
 
 T = TypeVar("T")

src/magentic/logger.py

@@ -1,6 +1,10 @@
 import logging
 
+import logfire_api
+
 logger = logging.getLogger("magentic")
 # Set default log level to WARNING so INFO logs must be explicitly enabled
 if logger.level == logging.NOTSET:
     logger.setLevel(logging.WARNING)
+
+logfire = logfire_api.Logfire(otel_scope="magentic")  # TODO: Pass version here too

src/magentic/prompt_chain.py

@@ -8,11 +8,10 @@
     cast,
 )
 
-import logfire_api as logfire
-
 from magentic.chat import Chat
 from magentic.chat_model.base import ChatModel
 from magentic.function_call import FunctionCall
+from magentic.logger import logfire
 from magentic.prompt_function import AsyncPromptFunction, PromptFunction
 
 P = ParamSpec("P")

src/magentic/prompt_function.py

@@ -14,12 +14,11 @@
     overload,
 )
 
-import logfire_api as logfire
-
 from magentic.backend import get_chat_model
 from magentic.chat_model.base import ChatModel
 from magentic.chat_model.message import UserMessage
 from magentic.chat_model.retry_chat_model import RetryChatModel
+from magentic.logger import logfire
 from magentic.typing import split_union_type
 
 P = ParamSpec("P")