kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

examples: fix gpt-oss-tools-stream for adding toolcalls (#568)
Some checks failed
test / test (push) Has been cancelled
Details
test / lint (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Parth Sareen 2025-08-21 13:44:59 -07:00 committed by GitHub
parent c8392d6524
commit da79e987f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/gpt-oss-tools-stream.py
View File

@ -55,7 +55,7 @@ client = Client(
# host="https://ollama.com", headers={'Authorization': (os.getenv('OLLAMA_API_KEY'))}
)
model = 'gpt-oss:120b'
model = 'gpt-oss:20b'
# gpt-oss can call tools while "thinking"
# a loop is needed to call the tools and get the results
final = True
@ -81,7 +81,7 @@ while True:
thinking += chunk.message.thinking
print(chunk.message.thinking, end='', flush=True)
if thinking != '' or content != '':
if thinking != '' or content != '' or len(tool_calls) > 0:
messages.append({'role': 'assistant', 'thinking': thinking, 'content': content, 'tool_calls': tool_calls})
print()