Saved result name in LLMResult

llmclient/llms.py

@@ -344,7 +344,7 @@ async def call(  # noqa: C901, PLR0915
             result.seconds_to_last_token = (
                 asyncio.get_running_loop().time() - start_clock
             )
-
+            result.name = name
             if self.llm_result_callback:
                 possibly_awaitable_result = self.llm_result_callback(result)
                 if isawaitable(possibly_awaitable_result):

tests/test_llms.py

@@ -104,7 +104,8 @@ async def test_call(self, config: dict[str, Any]) -> None:
             Message(role="system", content="Respond with single words."),
             Message(role="user", content="What is the meaning of the universe?"),
         ]
-        results = await llm.call(messages)
+        result_name = f"test-{llm.name}"
+        results = await llm.call(messages, name=result_name)
         assert isinstance(results, list)
 
         result = results[0]
@@ -116,7 +117,7 @@ async def test_call(self, config: dict[str, Any]) -> None:
         assert result.prompt[1].content
         assert result.text
         assert result.logprob is None or result.logprob <= 0
-
+        assert result.name == result_name
         result = await llm.call_single(messages)
         assert isinstance(result, LLMResult)