Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion m_flow/adapters/cache/cache_db_interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ class CacheDBInterface(ABC):
Subclasses must implement lock primitives and Q/A session storage.
"""

__slots__ = ("_host", "_port", "_key", "_lock")
__slots__ = ("_host", "_key", "_lock", "_port")

def __init__(self, host: str, port: int, lock_key: str) -> None:
self._host = host
Expand Down
8 changes: 4 additions & 4 deletions m_flow/adapters/vector/embeddings/FastembedEmbeddingEngine.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,13 +41,13 @@ class FastembedEmbeddingEngine(EmbeddingEngine):
"""

__slots__ = (
"model",
"dimensions",
"max_completion_tokens",
"batch_size",
"dimensions",
"embedding_model",
"tokenizer",
"max_completion_tokens",
"mock",
"model",
"tokenizer",
)

def __init__(
Expand Down
8 changes: 4 additions & 4 deletions m_flow/adapters/vector/embeddings/OllamaEmbeddingEngine.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,14 +44,14 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
"""

__slots__ = (
"model",
"batch_size",
"dimensions",
"max_completion_tokens",
"endpoint",
"hf_tokenizer_name",
"batch_size",
"tokenizer",
"max_completion_tokens",
"mock",
"model",
"tokenizer",
)

def __init__(
Expand Down
2 changes: 1 addition & 1 deletion m_flow/knowledge/graph_ops/m_flow_graph/MemoryGraph.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ class MemoryGraph(MemoryAbstractGraph):
``directed`` flag controls whether edge semantics are symmetric.
"""

__slots__ = ("nodes", "edges", "directed")
__slots__ = ("directed", "edges", "nodes")

nodes: Dict[str, Node]
edges: List[Edge]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ class Node:
Default cost when this node was **not** hit by any vector search.
"""

__slots__ = ("id", "attributes", "_neighbours", "_edges", "status")
__slots__ = ("_edges", "_neighbours", "attributes", "id", "status")

def __init__(
self,
Expand Down Expand Up @@ -134,7 +134,7 @@ class Edge:
Default cost when this edge was **not** matched by vector search.
"""

__slots__ = ("node1", "node2", "attributes", "directed", "status")
__slots__ = ("attributes", "directed", "node1", "node2", "status")

def __init__(
self,
Expand Down
2 changes: 1 addition & 1 deletion m_flow/llm/tokenizer/Gemini/adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ class GeminiTokenizer(TokenizerInterface):
Upper bound on output tokens for downstream callers.
"""

__slots__ = ("llm_model", "max_completion_tokens", "_api_client")
__slots__ = ("_api_client", "llm_model", "max_completion_tokens")

def __init__(
self,
Expand Down
2 changes: 1 addition & 1 deletion m_flow/llm/tokenizer/Mistral/adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ class MistralTokenizer(TokenizerInterface):
Downstream hint for maximum generation length.
"""

__slots__ = ("model", "max_completion_tokens", "_backing_tokenizer")
__slots__ = ("_backing_tokenizer", "max_completion_tokens", "model")

def __init__(
self,
Expand Down
2 changes: 1 addition & 1 deletion m_flow/pipeline/tasks/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ class Stage:
Keyword arguments merged into every invocation of *fn*.
"""

__slots__ = ("_fn", "_defaults", "_kw", "_cfg", "_kind", "_batch")
__slots__ = ("_batch", "_cfg", "_defaults", "_fn", "_kind", "_kw")

def __init__(
self,
Expand Down
2 changes: 1 addition & 1 deletion m_flow/shared/files/storage/FileBufferedReader.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
class FileBufferedReader(BufferedReader):
"""BufferedReader that exposes an explicit ``name`` property."""

__slots__ = ("_wrapped", "_label")
__slots__ = ("_label", "_wrapped")

def __init__(self, raw_stream, name: str) -> None:
super().__init__(raw_stream)
Expand Down
Loading