Update README.md

README.md

@@ -115,9 +115,34 @@ print(tokenizer.decode(r[0]))
 ```
 
 ```python
-f = { "name": "get_exchange_rate", "description": "Get the exchange rate between two currencies", "parameters": { "type": "object", "properties": { "base_currency": { "type": "string", "description": "The currency to convert from" }, "target_currency": { "type": "string", "description": "The currency to convert to" } }, "required": [ "base_currency", "target_currency" ] } }
+f = {
+    'name': 'parse_entities',
+    'description': 'extract entities from the text',
+    'parameters': {
+        'type': 'object',
+        'properties': {
+            'drink': {
+                'type': 'string',
+                'description': 'drink name',
+            },
+            'event': {
+                'type': 'string',
+                'description': 'event name',
+            },
+            'person_name': {
+                'type': 'string',
+                'description': 'person name',
+            }
+        },
+        'required': [
+            'drink',
+            'event',
+            'person_name'
+        ]
+    }
+}
 messages = [
-    {'role': 'user', 'content': 'tolong tukar 10 ringgit ke usd'}
+    {'role': 'user', 'content': 'nama saya husein bin zolkepli, saya sekarang berada di jomheboh 2023 sambil minum starbucks'}
 ]
 prompt = parse_mistral_chat(messages, function_call = [f])
 inputs = tokenizer([prompt], return_tensors='pt', add_special_tokens=False).to('cuda')
@@ -138,28 +163,34 @@ print(tokenizer.decode(r[0]))
 <s> 
 [FUNCTIONCALL]
 {
-    "name": "get_exchange_rate",
-    "description": "Get the exchange rate between two currencies",
+    "name": "parse_entities",
+    "description": "extract entities from the text",
     "parameters": {
         "type": "object",
         "properties": {
-            "base_currency": {
+            "drink": {
+                "type": "string",
+                "description": "drink name"
+            },
+            "event": {
                 "type": "string",
-                "description": "The currency to convert from"
+                "description": "event name"
             },
-            "target_currency": {
+            "person_name": {
                 "type": "string",
-                "description": "The currency to convert to"
+                "description": "person name"
             }
         },
         "required": [
-            "base_currency",
-            "target_currency"
+            "drink",
+            "event",
+            "person_name"
         ]
     }
 }
-[INST] tolong tukar 10 ringgit ke usd [/INST] <functioncall> {"name": "get_exchange_rate", "arguments": '{"base_currency": "MYR", "target_currency": "USD"}'} 
-
-
-<functioncall> {"exchange_rate": 0.1457}</s>
+[INST] nama saya husein bin zolkepli, saya sekarang berada di jomheboh 2023 sambil minum starbucks [/INST] <functioncall> {"name": "parse_entities", "arguments": '{
+  "drink": "Starbucks",
+  "event": "Jom Heboh 2023",
+  "person_name": "Husein Bin Zolkepli"
+}'}</s>
 ```