OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-21 01:31:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Ignore some fields while hashing the data for steps

Browse Source
This commit is contained in:
Zvonimir Sabljic
2023-08-09 09:15:37 +02:00
parent c907f8b9f3
commit 94a220a08e
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
euclid/const/common.py
View File

@@ -26,4 +26,6 @@ IGNORE_FOLDERS = [
'venv',
'dist',
'build',
]
]
PROMPT_DATA_TO_IGNORE = {'directory_tree', 'name'}

5
euclid/database/database.py
View File

@@ -3,6 +3,7 @@ from peewee import *
from termcolor import colored
from functools import reduce
import operator
from const.common import PROMPT_DATA_TO_IGNORE
from utils.utils import hash_data
from database.models.components.base_models import database
@@ -141,7 +142,7 @@ def save_development_step(app_id, prompt_path, prompt_data, llm_req_num, message
app = get_app(app_id)
hash_id = hash_data({
'prompt_path': prompt_path,
'prompt_data': {k: v for k, v in prompt_data.items() if k not in {"directory_tree"}},
'prompt_data': {} if prompt_data is None else {k: v for k, v in prompt_data.items() if k not in PROMPT_DATA_TO_IGNORE},
'llm_req_num': llm_req_num
})
try:
@@ -245,7 +246,7 @@ def get_user_input_from_hash_id(project, query):
def get_development_step_from_hash_id(app_id, prompt_path, prompt_data, llm_req_num):
hash_id = hash_data({
'prompt_path': prompt_path,
'prompt_data': {k: v for k, v in prompt_data.items() if k not in {"directory_tree"}},
'prompt_data': {} if prompt_data is None else {k: v for k, v in prompt_data.items() if k not in PROMPT_DATA_TO_IGNORE},
'llm_req_num': llm_req_num
})
try: