eliott/ollama-python
1
0
Fork 0
mirror of https://github.com/kharonsec/ollama-python synced 2026-04-25 15:14:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

examples: update tools.py to handle type casting from tool arguments (#443)

Browse Source 
---------

Co-authored-by: Parth Sareen <parth.sareen@ollama.com>
This commit is contained in:
ddbit
2025-02-14 18:50:47 +01:00
committed by GitHub
parent 8d0d0e483d
commit aec125c773
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/tools.py
View File

@@ -12,14 +12,19 @@ def add_two_numbers(a: int, b: int) -> int:
Returns:
int: The sum of the two numbers
"""
return a + b
# The cast is necessary as returned tool call arguments don't always conform exactly to schema
# E.g. this would prevent "what is 30 + 12" to produce '3012' instead of 42
return int(a) + int(b)
def subtract_two_numbers(a: int, b: int) -> int:
"""
Subtract two numbers
"""
return a - b
# The cast is necessary as returned tool call arguments don't always conform exactly to schema
return int(a) - int(b)
# Tools can still be manually defined and passed into chat