gaurv007 commited on
Commit
e8b127a
·
verified ·
1 Parent(s): 06c179a

Upload alpha_factory/infra/__init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. alpha_factory/infra/__init__.py +7 -1
alpha_factory/infra/__init__.py CHANGED
@@ -2,5 +2,11 @@
2
  from .llm_client import LLMClient
3
  from .factor_store import FactorStore
4
  from .wq_client import BrainClient
 
 
5
 
6
- __all__ = ["LLMClient", "FactorStore", "BrainClient"]
 
 
 
 
 
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
+ ]