mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
50bef87da7
Metadata for the Linear UI is now sneakily provided via a `MetadataAccumulator` node, which the client populates / hooks up while building the graph.
Additionally, we provide the unexpanded graph with the metadata API response.
Both of these are embedded into the PNGs.
- Remove `metadata` from `ImageDTO`
- Split up the `images/` routes to accomodate this; metadata is only retrieved per-image
- `images/{image_name}` now gets the DTO
- `images/{image_name}/metadata` gets the new metadata
- `images/{image_name}/full` gets the full-sized image file
- Remove old metadata service
- Add `MetadataAccumulator` node, `CoreMetadataField`, hook up to `LatentsToImage` node
- Add `get_raw()` method to `ItemStorage`, retrieves the row from DB as a string, no pydantic parsing
- Update `images`related services to handle storing and retrieving the new metadata
- Add `get_metadata_graph_from_raw_session` which extracts the `graph` from `session` without needing to hydrate the session in pydantic, in preparation for providing it as metadata; also removes all references to the `MetadataAccumulator` node
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import json
|
|
from typing import Optional
|
|
|
|
from pydantic import ValidationError
|
|
|
|
from invokeai.app.services.graph import Edge
|
|
|
|
|
|
def get_metadata_graph_from_raw_session(session_raw: str) -> Optional[dict]:
|
|
"""
|
|
Parses raw session string, returning a dict of the graph.
|
|
|
|
Only the general graph shape is validated; none of the fields are validated.
|
|
|
|
Any `metadata_accumulator` nodes and edges are removed.
|
|
|
|
Any validation failure will return None.
|
|
"""
|
|
|
|
graph = json.loads(session_raw).get("graph", None)
|
|
|
|
# sanity check make sure the graph is at least reasonably shaped
|
|
if (
|
|
type(graph) is not dict
|
|
or "nodes" not in graph
|
|
or type(graph["nodes"]) is not dict
|
|
or "edges" not in graph
|
|
or type(graph["edges"]) is not list
|
|
):
|
|
# something has gone terribly awry, return an empty dict
|
|
return None
|
|
|
|
try:
|
|
# delete the `metadata_accumulator` node
|
|
del graph["nodes"]["metadata_accumulator"]
|
|
except KeyError:
|
|
# no accumulator node, all good
|
|
pass
|
|
|
|
# delete any edges to or from it
|
|
for i, edge in enumerate(graph["edges"]):
|
|
try:
|
|
# try to parse the edge
|
|
Edge(**edge)
|
|
except ValidationError:
|
|
# something has gone terribly awry, return an empty dict
|
|
return None
|
|
|
|
if (
|
|
edge["source"]["node_id"] == "metadata_accumulator"
|
|
or edge["destination"]["node_id"] == "metadata_accumulator"
|
|
):
|
|
del graph["edges"][i]
|
|
|
|
return graph
|