docs: update ai21 docs and readme with jamba 1.5 and tool calling

run-llama · Sep 10, 2024 · e9ff660 · e9ff660
1 parent b8f0837
commit e9ff660
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/docs/docs/examples/llm/ai21.ipynb b/docs/docs/examples/llm/ai21.ipynb
@@ -368,7 +368,7 @@
     "\n",
     "\n",
     "def multiply(a: int, b: int) -> int:\n",
-    "    \"\"\"Multiple two integers and returns the result integer\"\"\"\n",
+    "    \"\"\"Multiply two integers and returns the result integer\"\"\"\n",
     "    return a * b\n",
     "\n",
     "def subtract(a: int, b: int) -> int:\n",
@@ -398,7 +398,7 @@
     ")\n",
     "agent = agent_worker.as_agent()\n",
     "\n",
-    "response = agent.chat(\"My friend moses had 10 apples. He ate 5 apples in the morning. Then he found a box with 25 apples. he divided all his apples between his 5 friends. How many apples did each friend got?\")"
+    "response = agent.chat(\"My friend Moses had 10 apples. He ate 5 apples in the morning. Then he found a box with 25 apples. He divided all his apples between his 5 friends. How many apples did each friend get?\")"
    ]
   }
  ],

diff --git a/llama-index-integrations/llms/llama-index-llms-ai21/README.md b/llama-index-integrations/llms/llama-index-llms-ai21/README.md
@@ -179,7 +179,7 @@ from llama_index.core.tools import FunctionTool
 
 
 def multiply(a: int, b: int) -> int:
-    """Multiple two integers and returns the result integer"""
+    """Multiply two integers and returns the result integer"""
     return a * b
 
 def subtract(a: int, b: int) -> int:
@@ -211,6 +211,6 @@ agent_worker = FunctionCallingAgentWorker.from_tools(
 )
 agent = agent_worker.as_agent()
 
-response = agent.chat("My friend moses had 10 apples. He ate 5 apples in the morning. Then he found a box with 25 apples."
-                      "he divided all his apples between his 5 friends. How many apples did each friend got?")
+response = agent.chat("My friend Moses had 10 apples. He ate 5 apples in the morning. Then he found a box with 25 apples."
+                      "He divided all his apples between his 5 friends. How many apples did each friend get?")
 ```