Update README.md

README.md

@@ -70,8 +70,87 @@ print(result)
 
 ## Feature: Visual Instruction Following
 
+```python
+
+```
+
+
 ## Feature: Function Calling
 
+```python
+import json
+
+functions = [
+    {
+      "name": "get_current_weather",
+      "description": "Get the current weather in a given location",
+      "parameters": {
+        "type": "object",
+        "properties": {
+          "location": {
+            "type": "string",
+            "description": "The city and state, e.g. San Francisco, CA"
+          },
+          "unit": {
+            "type": "string",
+            "enum": ["celsius", "fahrenheit"]
+          }
+        },
+        "required": ["location"]
+      }
+    }
+]
+
+def fake_get_current_weather(location, unit=None):
+    return {'temperature': 30}
+
+mapping = {
+    'get_current_weather': fake_get_current_weather
+}
+
+# stage 1: query
+conversations = [
+    {"role": "user", "content": "請問台北目前溫度是攝氏幾度？"},
+]
+
+prompt = prompt_engine.get_prompt(conversations, functions=functions)
+
+output_str = _inference(prompt, tokenizer, model, generation_config)
+result = prompt_engine.parse_generated_str(output_str)
+
+print(result)
+# {'role': 'assistant', 'tool_calls': [{'id': 'call_0bcY2wePCVTg14Q6Xor93fHz', 'type': 'function', 'function': {'name': 'get_current_weather', 'arguments': '{"location": "台北", "unit": "celsius"}'}}]}
+```
+
+
+```python
+# stage 2: execute called functions
+conversations.append(result)
+
+tool_call = result['tool_calls'][0]
+func_name = tool_call['function']['name']
+func = mapping[func_name]
+arguments = json.loads(tool_call['function']['arguments'])
+called_result = func(**arguments)
+
+# stage 3: put executed results
+conversations.append(
+    {
+        'role': 'tool',
+        'tool_call_id': tool_call['id'],
+        'name': func_name,
+        'content': json.dumps(called_result)
+    }
+)
+
+prompt = prompt_engine.get_prompt(conversations, functions=functions)
+
+output_str2 = _inference(prompt, tokenizer, model, generation_config)
+result2 = prompt_engine.parse_generated_str(output_str2)
+print(result2)
+# {'role': 'assistant', 'content': '台北目前的溫度是攝氏30度。'}
+```
+
 ## Citation
 
 ```