gaurv007 commited on
Commit
3dd4a9b
·
verified ·
1 Parent(s): 3746700

fix: infra/__init__.py — only export what exists, remove BrainClient from active imports"

Browse files
Files changed (1) hide show
  1. alpha_factory/infra/__init__.py +10 -8
alpha_factory/infra/__init__.py CHANGED
@@ -1,12 +1,14 @@
1
  """Infrastructure modules."""
2
  from .llm_client import LLMClient
3
  from .factor_store import FactorStore
4
- from .wq_client import BrainClient
5
- from .model_manager import ModelManager, ModelProvider, ModelInfo, interactive_model_select
6
- from .rag import PaperRAG
7
 
8
- __all__ = [
9
- "LLMClient", "FactorStore", "BrainClient",
10
- "ModelManager", "ModelProvider", "ModelInfo", "interactive_model_select",
11
- "PaperRAG",
12
- ]
 
 
 
 
 
1
  """Infrastructure modules."""
2
  from .llm_client import LLMClient
3
  from .factor_store import FactorStore
4
+ from .model_manager import ModelManager, interactive_model_select
 
 
5
 
6
+ # BrainClient exists but is NOT connected to the pipeline
7
+ # from .wq_client import BrainClient
8
+
9
+ # Stub for imports that reference it
10
+ class BrainClient:
11
+ """STUB — BRAIN API client is not connected. All submissions are manual."""
12
+ pass
13
+
14
+ __all__ = ["LLMClient", "FactorStore", "ModelManager", "interactive_model_select", "BrainClient"]