Update README.md

README.md

@@ -20,3 +20,77 @@ language:
 This gemma2 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
 [<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+
+-- teste esse codigo com o prompt modificado para realizar chamadas de funcao
+
+from langchain.agents import AgentType, initialize_agent, load_tools
+from langchain.agents import AgentExecutor
+from langchain.agents import tool, load_tools, create_react_agent
+from langchain import hub
+import os
+from langchain_ollama.llms import OllamaLLM
+from langchain.prompts import PromptTemplate
+
+# Criar tool personalizada
+@tool
+def get_word_length(word: str) -> int:
+    """Returns the length of a word."""
+    return len(word)
+
+MODEL = "hf.co/vinimuchulski/GEMMA-2-2B-it-GGUF-function_calling:latest"
+
+custom_react_prompt = PromptTemplate(
+    input_variables=["input", "agent_scratchpad", "tools", "tool_names"],
+    template="""Answer the following questions as best you can. You have access to the following tools:
+
+{tools}
+
+Use the following format:
+
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action: the action to take, should be one of [{tool_names}]
+Action Input: the input to the action, formatted as a string
+Observation: the result of the action
+Thought: I now know the final answer
+Final Answer: the final answer to the original input question
+
+Example:
+Question: What is the length of the word "hello"?
+Thought: I need to use the get_word_length tool to calculate the length of the word "hello".
+Action: get_word_length
+Action Input: "hello"
+Observation: 5
+Thought: I now know the length of the word "hello" is 5.
+Final Answer: 5
+
+Begin!
+
+Question: {input}
+Thought: {agent_scratchpad}"""
+)
+
+# Formatar as ferramentas para o prompt
+tools = [get_word_length]
+tools_str = "\n".join([f"{tool.name}: {tool.description}" for tool in tools])
+tool_names = ", ".join([tool.name for tool in tools])
+
+# Criar o agente com o prompt personalizado
+agent = create_react_agent(
+    tools=tools,
+    llm=llm,
+    prompt=custom_react_prompt.partial(tools=tools_str, tool_names=tool_names),
+)
+
+# Criar o executor
+agent_executor = AgentExecutor(
+    agent=agent,
+    tools=tools,
+    verbose=True,
+    handle_parsing_errors=True
+)
+
+# Testar
+question = "What is the length of the word PythonDanelonAugustoTrajanoRomanovCzarVespasianoDiocleciano ?"
+response = agent_executor.invoke({"input": question})
+print(response)