mirror of
https://github.com/Significant-Gravitas/AutoGPT-Code-Ability.git
synced 2026-11-10 08:01:27 -10:00
Swifty
b472a58422
* PR Coments * Fixed db issue * Add async main function to database.py * fixed linting errors * fixed linting errors * fixed linting errors * fixed linting errors * more linting * formatting
29 lines
754 B
Python
29 lines
754 B
Python
import prisma.models
|
|
|
|
INCLUDE_TYPE = {"include": {"RelatedTypes": {"include": {"Fields": True}}}}
|
|
|
|
INCLUDE_FIELD = {"include": {"Fields": {"include": {"RelatedTypes": True}}}}
|
|
|
|
INCLUDE_API_ROUTE = {
|
|
"include": {
|
|
"RequestObject": INCLUDE_FIELD,
|
|
"ResponseObject": INCLUDE_FIELD,
|
|
}
|
|
}
|
|
|
|
INCLUDE_FUNC = {
|
|
"include": {
|
|
"FunctionArgs": INCLUDE_TYPE,
|
|
"FunctionReturn": INCLUDE_TYPE,
|
|
"DatabaseSchema": {"include": {"DatabaseTables": True}},
|
|
}
|
|
}
|
|
|
|
|
|
def get_database_schema(spec: prisma.models.Specification) -> str:
|
|
return (
|
|
"\n\n".join([t.definition for t in spec.DatabaseSchema.DatabaseTables])
|
|
if spec and spec.DatabaseSchema and spec.DatabaseSchema.DatabaseTables
|
|
else ""
|
|
)
|