blob: 5a7078863bb83cbdf5c297c614fba7a7086ca0ff (
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
|
from langchain_ollama import ChatOllama
from langchain.agents import create_agent
from langchain_community.tools.file_management import FileSearchTool
GEMMA3 = "gemma3"
LLAMA3 = "llama3.1:8b"
model = ChatOllama(
model=LLAMA3,
validate_model_on_init=True,
temperature=0.8,
num_predict=256,
)
agent = create_agent(
model,
tools=[
FileSearchTool(),
],
system_prompt="You are a helpful assistent.",
)
result = agent.invoke(
{
"messages": [
{
"role": "user",
"content": "What files match *.py in .?",
}
]
}
)
for message in result["messages"]:
print(message.content)
|
