Skip to content

Commit

Permalink
Merge pull request #1138 from gpt-engineer-org/1128-applying-diffs-fa…
Browse files Browse the repository at this point in the history
…iling-silently

Enhance Diff Application and Error Transparency
  • Loading branch information
similato87 authored May 11, 2024
2 parents 5a5b785 + 46cfacd commit 612abb1
Show file tree
Hide file tree
Showing 4 changed files with 110 additions and 1 deletion.
3 changes: 2 additions & 1 deletion gpt_engineer/core/diff.py
Original file line number Diff line number Diff line change
Expand Up @@ -354,7 +354,8 @@ def validate_and_correct(self, lines_dict: dict) -> List[str]:
}
is_valid = hunk.validate_and_correct(cut_lines_dict, problems)
if not is_valid and len(problems) > 0:
print(f"Invalid hunk: {hunk.hunk_to_string()}")
for idx, val in enumerate(problems):
print(f"\nInvalid Hunk NO.{idx}---\n{val}\n---")
self.hunks.remove(hunk)
# now correct the numbers, assuming the start line pre-edit has been fixed
hunk.hunk_len_pre_edit = (
Expand Down
79 changes: 79 additions & 0 deletions tests/core/improve_function_test_cases/theo_case_chat
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
To add an entrypoint for running the API through the command line, we will create a new Python script that can be executed directly. This script will set up and run the FastAPI application using Uvicorn, which is an ASGI server. This approach allows you to run the server from the command line without directly using the `uvicorn` command.

We will create a new file named `run.py` in the root directory. This file will import the FastAPI application from `main.py` and will use Uvicorn to run it.

Here's the implementation:

```diff
--- /dev/null
+++ run.py
@@ -0,0 +1,10 @@
+import uvicorn
+from main import app
+
+# This script can be used to run the FastAPI application from the command line.
+if __name__ == "__main__":
+ uvicorn.run(app, host="0.0.0.0", port=8000)
```

This script checks if it is being run as the main module and then starts the Uvicorn server with the FastAPI app. The server listens on all interfaces (`0.0.0.0`) and port `8000`.

Additionally, we need to update the Dockerfile to ensure that the new script is correctly utilized when the Docker container is run. We will change the CMD directive to use `run.py` instead of directly calling `uvicorn`.

```diff
--- dockerfile
+++ dockerfile
@@ -19,3 +19,3 @@
-# Run app.py when the container launches
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
+# Run run.py when the container launches
+CMD ["python", "run.py"]
```

This change in the Dockerfile ensures that when the Docker container is started, it uses the new `run.py` script to launch the application. This script sets up the Uvicorn server with the specified host and port settings.

With these changes, you can now run the API both directly from the command line by executing `python run.py` and through Docker using the updated Dockerfile configuration. This setup provides flexibility for different deployment scenarios.Invalid hunk: @@ -19,3 +19,3 @@
-# Run app.py when the container launches
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
+# Run run.py when the container launches
+CMD ["python", "run.py"]

To add an entrypoint for running the API through the command line, we will create a new Python script that can be executed directly. This script will set up and run the FastAPI application using Uvicorn, which is an ASGI server. This approach allows you to run the server from the command line without directly using the `uvicorn` command.

We will create a new file named `run.py` in the root directory. This file will import the FastAPI application from `main.py` and will use Uvicorn to run it.

Here's the implementation:

```diff
--- /dev/null
+++ run.py
@@ -0,0 +1,10 @@
+import uvicorn
+from main import app
+
+# This script can be used to run the FastAPI application from the command line.
+if __name__ == "__main__":
+ uvicorn.run(app, host="0.0.0.0", port=8000)
```

This script checks if it is being run as the main module and then starts the Uvicorn server with the FastAPI app. The server listens on all interfaces (`0.0.0.0`) and port `8000`.

Additionally, we need to update the Dockerfile to ensure that the new script is correctly utilized when the Docker container is run. We will change the CMD directive to use `run.py` instead of directly calling `uvicorn`.

```diff
--- dockerfile
+++ dockerfile
@@ -19,3 +19,3 @@
-# Run app.py when the container launches
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
+# Run run.py when the container launches
+CMD ["python", "run.py"]
```

This change in the Dockerfile ensures that when the Docker container is started, it uses the new `run.py` script to launch the application. This script sets up the Uvicorn server with the specified host and port settings.

With these changes, you can now run the API both directly from the command line by executing `python run.py` and through Docker using the updated Dockerfile configuration. This setup provides flexibility for different deployment scenarios.Invalid hunk: @@ -19,3 +19,3 @@
-# Run app.py when the container launches
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
+# Run run.py when the container launches
+CMD ["python", "run.py"]
20 changes: 20 additions & 0 deletions tests/core/improve_function_test_cases/theo_case_code
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
# Use an official Python runtime as a parent image
FROM python:3.9-slim

# Set the working directory in the container
WORKDIR /usr/src/app

# Copy the current directory contents into the container at /usr/src/app
COPY . /usr/src/app

# Install any needed packages specified in requirements.txt
RUN pip install --no-cache-dir -r requirements.txt

# Make port 80 available to the world outside this container
EXPOSE 80

# Define environment variable
ENV NAME World

# Run app.py when the container launches
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
9 changes: 9 additions & 0 deletions tests/core/test_salvage_correct_hunks.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,15 @@ def test_create_two_new_files():
salvage_correct_hunks(message_builder("create_two_new_files_chat"), files, memory)


def test_theo_case():
files = FilesDict({"dockerfile": get_file_content("theo_case_code")})
updated_files, _ = salvage_correct_hunks(
message_builder("theo_case_chat"), files, memory
)
print(updated_files["dockerfile"])
print(updated_files["run.py"])


def test_clean_up_folder(clean_up_folder):
# The folder should be deleted after the test is run
assert True
Expand Down

0 comments on commit 612abb1

Please sign in to comment.