Nail/contracts-flask
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(ревью#2): db_docs→_db + classify_raw в MemRepo + try/except в build_classify_prompt

Browse Source
This commit is contained in:
naeel
2026-06-28 09:40:00 +04:00
parent 8a804da494
commit d8e8b35092
4 changed files with 89 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
deploy/repository.py
+12 -10
View File
@@ -31,7 +31,8 @@ class Repository(Protocol):
def set_classification(self, doc_id: str, doc_type: str, own_number: str = None,
parent_number: str = None, doc_date: str = None,
counterparty: str = None) -> None:
counterparty: str = None,
classify_raw: str = None, classify_input: str = None) -> None:
"""Сохранить результат классификации."""
...
@@ -107,10 +108,12 @@ class PgRepository:
documents.set_error(doc_id, error)
def set_classification(self, doc_id, doc_type, own_number=None, parent_number=None,
doc_date=None, counterparty=None):
doc_date=None, counterparty=None,
classify_raw=None, classify_input=None):
from db import documents
documents.set_classification(doc_id, doc_type, own_number, parent_number,
doc_date, counterparty)
doc_date, counterparty,
classify_raw=classify_raw, classify_input=classify_input)
def set_classify_garbage(self, doc_id, reason=""):
from db import documents
@@ -199,15 +202,14 @@ class MemRepository:
self.documents[doc_id]["error_message"] = error
def set_classification(self, doc_id, doc_type, own_number=None, parent_number=None,
doc_date=None, counterparty=None):
doc_date=None, counterparty=None,
classify_raw=None, classify_input=None):
if doc_id in self.documents:
d = self.documents[doc_id]
d["doc_type"] = doc_type
d["own_number"] = own_number
d["parent_number"] = parent_number
d["doc_date"] = doc_date
d["counterparty"] = counterparty
d["classify_status"] = "classified"
d.update({"doc_type": doc_type, "own_number": own_number,
"parent_number": parent_number, "doc_date": doc_date,
"counterparty": counterparty, "classify_raw": classify_raw,
"classify_input": classify_input, "classify_status": "classified"})
def set_classify_garbage(self, doc_id, reason=""):
if doc_id in self.documents: