How to Add a Docstring to Your Code
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
def add(a: int, b: int) -> int:
"""
This function takes two integers 'a' and 'b' and returns their sum.
Parameters:
a (int): The first integer.
b (int): The second integer.
Returns:
int: The sum of 'a' and 'b'.
"""
return a + bAPI Request
import cohere
co = cohere.ClientV2(api_key="<YOUR API KEY>")
response = co.chat(
model="command-a-plus-05-2026",
messages=[
{
"role": "user",
"content": """
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
```
""",
}
],
)
print(response.message.content[0].text)