codeflash-admin/codeflash-internal
1
0
Fork 0
mirror of https://github.com/codeflash-ai/codeflash-internal.git synced 2026-05-04 18:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Print optimized function for testing

Browse source
This commit is contained in:
afik.cohen 2023-10-20 16:53:08 -07:00
parent 068bf4d153
commit d13c8715b1
1 changed files with 16 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
codeflash/optimization/optimizer.py
View file

@ -4,7 +4,9 @@ from typing import List, Tuple, Union
import openai
def optimize_python_code(source_code: str, n: int = 1) -> List[Tuple[Union[str, None], Union[str, None]]]:
def optimize_python_code(
source_code: str, n: int = 1
) -> List[Tuple[Union[str, None], Union[str, None]]]:
"""
Optimize the given python code for performance using OpenAI's GPT-4 model.
@ -17,13 +19,15 @@ def optimize_python_code(source_code: str, n: int = 1) -> List[Tuple[Union[str,
"""
system_message = {
"role": "system",
"content": ("You are a computer programmer who writes really fast programs and is an expert in optimizing "
"runtime and memory requirements of a program by rewriting them. You don't rename function names "
"or change function signatures. The function return value should be exactly the same as before.")
"content": (
"You are a computer programmer who writes really fast programs and is an expert in optimizing "
"runtime and memory requirements of a program by rewriting them. You don't rename function names "
"or change function signatures. The function return value should be exactly the same as before."
),
}
user_message = {
"role": "user",
"content": f"Rewrite this python program to run faster.\n```python\n{source_code}\n```"
"content": f"Rewrite this python program to run faster.\n```python\n{source_code}\n```",
}
messages = [system_message, user_message]
@ -49,7 +53,9 @@ def optimize_python_code(source_code: str, n: int = 1) -> List[Tuple[Union[str,
if __name__ == "__main__":
optimize_python_code("""
print(
optimize_python_code(
"""
def sorter(arr):
n = len(arr)
for i in range(n):
@ -61,4 +67,7 @@ if __name__ == "__main__":
if swapped == False:
break
return arr
""", 3)
""",
3,
)
)