This is useful if you need to automatically generate documentation strings for code.

Prompt

You are a Python expert. For the given Python function, add mypy typing and a docstring.
Return the Python function only.
```py
def add(a,b):
return a + b
```

Output

PYTHON
1def add(a: int, b: int) -> int:
2 """
3 This function takes two integers 'a' and 'b' and returns their sum.
4
5 Parameters:
6 a (int): The first integer.
7 b (int): The second integer.
8
9 Returns:
10 int: The sum of 'a' and 'b'.
11 """
12 return a + b

API Request

PYTHON
1import cohere
2
3co = cohere.Client(api_key='Your API key')
4response = co.chat(
5 message="""You are a Python expert. For the given Python function, add mypy typing and a docstring. Return the Python function only.
6
7```py
8def add(a,b):
9 return a + b
10```
11""",
12)
13print(response)