diff --git a/src/aidgaf/aidgaf_server/idgaf.py b/src/aidgaf/aidgaf_server/idgaf.py index 9bd70be..d14a08e 100644 --- a/src/aidgaf/aidgaf_server/idgaf.py +++ b/src/aidgaf/aidgaf_server/idgaf.py @@ -91,7 +91,7 @@ def get_prompt(command) -> dict: print(my_prompt) the_data = DATA - the_data["prompt"] = my_prompt + the_data["messages"][-1]["content"] = my_prompt return the_data