Spaces:
Running
Running
Commit
·
f0fc928
1
Parent(s):
b522e95
ad
Browse files
agentgraph/shared/method_registry.py
CHANGED
|
@@ -23,6 +23,18 @@ class SchemaType(Enum):
|
|
| 23 |
|
| 24 |
# Method Registry - maps method names to their implementations and schemas
|
| 25 |
AVAILABLE_METHODS = {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
# Production method using reference-based schema
|
| 27 |
"production": {
|
| 28 |
"name": "Multi-Agent Knowledge Extractor",
|
|
@@ -150,7 +162,7 @@ AVAILABLE_METHODS = {
|
|
| 150 |
|
| 151 |
|
| 152 |
# Default method configuration
|
| 153 |
-
DEFAULT_METHOD = "
|
| 154 |
|
| 155 |
|
| 156 |
def get_available_methods() -> Dict[str, Dict[str, Any]]:
|
|
@@ -226,4 +238,4 @@ def get_method_description(method_name: str) -> str:
|
|
| 226 |
def validate_method_schema_compatibility(method_name: str, expected_schema: SchemaType) -> bool:
|
| 227 |
"""Validate that a method uses the expected schema type"""
|
| 228 |
method_schema = get_schema_for_method(method_name)
|
| 229 |
-
return method_schema == expected_schema if method_schema else False
|
|
|
|
| 23 |
|
| 24 |
# Method Registry - maps method names to their implementations and schemas
|
| 25 |
AVAILABLE_METHODS = {
|
| 26 |
+
# Production method: OpenAI Structured Outputs (simple, direct_call)
|
| 27 |
+
"openai_structured": {
|
| 28 |
+
"name": "OpenAI Structured Outputs",
|
| 29 |
+
"description": "Simple OpenAI structured outputs extractor using Pydantic models",
|
| 30 |
+
"method_type": MethodType.PRODUCTION,
|
| 31 |
+
"schema_type": SchemaType.REFERENCE_BASED,
|
| 32 |
+
"module_path": "agentgraph.methods.production.openai_structured_extractor",
|
| 33 |
+
"class_name": "OpenAIStructuredFactory",
|
| 34 |
+
"supported_features": ["structured_outputs", "direct_extraction"],
|
| 35 |
+
"processing_type": "direct_call"
|
| 36 |
+
},
|
| 37 |
+
|
| 38 |
# Production method using reference-based schema
|
| 39 |
"production": {
|
| 40 |
"name": "Multi-Agent Knowledge Extractor",
|
|
|
|
| 162 |
|
| 163 |
|
| 164 |
# Default method configuration
|
| 165 |
+
DEFAULT_METHOD = "openai_structured"
|
| 166 |
|
| 167 |
|
| 168 |
def get_available_methods() -> Dict[str, Dict[str, Any]]:
|
|
|
|
| 238 |
def validate_method_schema_compatibility(method_name: str, expected_schema: SchemaType) -> bool:
|
| 239 |
"""Validate that a method uses the expected schema type"""
|
| 240 |
method_schema = get_schema_for_method(method_name)
|
| 241 |
+
return method_schema == expected_schema if method_schema else False
|