How to Add a Docstring to Your Code (v1 API)

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

1 def 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

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

1	def 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

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