fix: modify params on ollama granite cookbook (#260)

Signed-off-by: MICHAEL DESMOND <mdesmond@us.ibm.com>
i-am-bee · Dec 18, 2024 · f6e42a7 · f6e42a7
1 parent 56045d6
commit f6e42a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/examples/agents/granite/README.md b/examples/agents/granite/README.md
@@ -99,7 +99,7 @@ ollama serve
 Run the [granite_wiki_bee](/examples/agents/granite/granite_wiki_bee.ts) agent:
 
 ```shell
-yarn run start examples/agents/granite/granite_wiki_bee.ts <<< "Who were the authors of the paper 'Attention is all you need' and how many citations does it have?"
+yarn run start examples/agents/granite/granite_wiki_bee.ts <<< "Who were the authors of the research paper 'Attention is all you need', how many citations does it have?"
 ```
 
 You will see the agent reasoning, calling the WikipediaTool and producing a final answer similar to the following:

diff --git a/examples/agents/granite/granite_wiki_bee.ts b/examples/agents/granite/granite_wiki_bee.ts
@@ -79,8 +79,7 @@ const llm = new OllamaChatLLM({
   modelId: "granite3.1-dense:8b",
   parameters: {
     temperature: 0,
-    num_ctx: 4096,
-    num_predict: 512,
+    num_predict: 2048,
   },
 });