blob: 463b887469c16269d0e94998ea7aeb527fd774bd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
from haystack_integrations.components.generators.ollama import OllamaChatGenerator
from haystack.core.component import component
from haystack.components.agents import Agent
from haystack.tools import Tool, tool
from typing import List
@tool
def validate_user(user_id: int, addresses: List[str]) -> str:
"""Validate user using historical addresses.
Args:
user_id (int): the user ID.
addresses (List[str]): Previous addresses as a list of strings.
"""
return f"User {user_id} validated with addresses: {addresses}"
llm = OllamaChatGenerator(model="qwen3:1.7b",
url="http://localhost:11434",
generation_kwargs={
"temperature": 0,
})
from haystack.dataclasses import ChatMessage, ChatRole
prompt = "Could you validate user 123? They previously lived at 123 Fake St in Boston MA and 234 Pretend Boulevard in Houston TX. Think before you call any tools"
agent = Agent(chat_generator=llm, tools=[validate_user])
result = agent.run([ChatMessage.from_user(prompt)])
print("--- Full Haystack Result ---")
print(result)
print("\n" + "="*40 + "\n")
print("--- Extracted Reasoning ---")
for message in result["messages"]:
if message.role == ChatRole.ASSISTANT:
for content_block in message.content:
if hasattr(content_block, 'text') and "<think>" in content_block.text:
reasoning = content_block.text.split("<think>")[1].split("</think>")[0]
print(reasoning.strip())
|
