Skip to content

Commit dd77506

Browse files
committed
fix(arkitect): fix built in tools
1 parent 7333634 commit dd77506

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

arkitect/core/component/tool/builtin_tools.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from pydantic import BaseModel
15+
from volcenginesdkarkruntime._models import BaseModel
1616

1717
from arkitect.core.client.http import default_ark_client
1818

@@ -34,7 +34,7 @@ async def link_reader(url_list: list[str]) -> dict:
3434
}
3535

3636
response = await client.post(path="/tools/execute", body=body, cast_to=BaseModel)
37-
return response
37+
return response.model_dump()
3838

3939

4040
async def calculator(input: str) -> dict:
@@ -53,4 +53,4 @@ async def calculator(input: str) -> dict:
5353
}
5454

5555
response = await client.post(path="/tools/execute", body=body, cast_to=BaseModel)
56-
return response
56+
return response.model_dump()

0 commit comments

Comments
 (0)