1 files changed, 37 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..5a70788
--- /dev/null
+++ b/main.py
@@ -0,0 +1,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)