mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-07 18:19:00 +00:00
Fix: replace new_data_artifact with new_artifact (available in a2a-sdk 1.0.0)
Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
@@ -14,7 +14,7 @@ from typing import TYPE_CHECKING, Any, ParamSpec, TypeVar, cast
|
|||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
from a2a.helpers.proto_helpers import (
|
from a2a.helpers.proto_helpers import (
|
||||||
new_data_artifact,
|
new_artifact,
|
||||||
new_text_artifact,
|
new_text_artifact,
|
||||||
new_text_message as new_agent_text_message,
|
new_text_message as new_agent_text_message,
|
||||||
)
|
)
|
||||||
@@ -286,9 +286,17 @@ def _create_result_artifact(
|
|||||||
"""
|
"""
|
||||||
artifact_name = f"result_{task_id}"
|
artifact_name = f"result_{task_id}"
|
||||||
if isinstance(result, dict):
|
if isinstance(result, dict):
|
||||||
return new_data_artifact(artifact_name, result)
|
from google.protobuf import struct_pb2
|
||||||
|
|
||||||
|
val = struct_pb2.Value()
|
||||||
|
val.struct_value.update(result)
|
||||||
|
return new_artifact([Part(data=val)], artifact_name)
|
||||||
if isinstance(result, BaseModel):
|
if isinstance(result, BaseModel):
|
||||||
return new_data_artifact(artifact_name, result.model_dump())
|
from google.protobuf import struct_pb2
|
||||||
|
|
||||||
|
val = struct_pb2.Value()
|
||||||
|
val.struct_value.update(result.model_dump())
|
||||||
|
return new_artifact([Part(data=val)], artifact_name)
|
||||||
return new_text_artifact(artifact_name, str(result))
|
return new_text_artifact(artifact_name, str(result))
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user