Skip to content

Commit 3b3f80c

Browse files
YoanSallamiYoanSallami
committed
Update API and clean up
1 parent e964d41 commit 3b3f80c

File tree

34 files changed

+460
-353
lines changed

34 files changed

+460
-353
lines changed

‎synalinks/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
from synalinks.api import config
6060
from synalinks.api import datasets
6161
from synalinks.api import default_prompt_template
62+
from synalinks.api import disable_telemetry
6263
from synalinks.api import embedding_models
6364
from synalinks.api import initializers
6465
from synalinks.api import is_chat_message

‎synalinks/api/__init__.py‎

Lines changed: 105 additions & 88 deletions
Original file line numberDiff line numberDiff line change
@@ -4,92 +4,109 @@
44
since your modifications would be overwritten.
55
"""
66

7-
from synalinks.api import backend
8-
from synalinks.api import callbacks
9-
from synalinks.api import config
10-
from synalinks.api import datasets
11-
from synalinks.api import embedding_models
12-
from synalinks.api import initializers
13-
from synalinks.api import language_models
14-
from synalinks.api import metrics
15-
from synalinks.api import modules
16-
from synalinks.api import ops
17-
from synalinks.api import optimizers
18-
from synalinks.api import programs
19-
from synalinks.api import rewards
20-
from synalinks.api import saving
21-
from synalinks.api import tree
22-
from synalinks.api import utils
23-
from synalinks.src.backend import DataModel
24-
from synalinks.src.backend import name_scope
25-
from synalinks.src.backend.common.global_state import clear_session
26-
from synalinks.src.backend.common.json_data_model import JsonDataModel
27-
from synalinks.src.backend.common.stateless_scope import StatelessScope
28-
from synalinks.src.backend.common.symbolic_data_model import SymbolicDataModel
29-
from synalinks.src.backend.common.symbolic_scope import SymbolicScope
30-
from synalinks.src.backend.config import synalinks_home
31-
from synalinks.src.backend.pydantic.base import ChatMessage
32-
from synalinks.src.backend.pydantic.base import ChatMessages
33-
from synalinks.src.backend.pydantic.base import ChatRole
34-
from synalinks.src.backend.pydantic.base import EmbeddedEntity
35-
from synalinks.src.backend.pydantic.base import Embeddings
36-
from synalinks.src.backend.pydantic.base import Entities
37-
from synalinks.src.backend.pydantic.base import Entity
38-
from synalinks.src.backend.pydantic.base import GenericInputs
39-
from synalinks.src.backend.pydantic.base import GenericIO
40-
from synalinks.src.backend.pydantic.base import GenericOutputs
41-
from synalinks.src.backend.pydantic.base import GenericResult
42-
from synalinks.src.backend.pydantic.base import Instructions
43-
from synalinks.src.backend.pydantic.base import KnowledgeGraph
44-
from synalinks.src.backend.pydantic.base import Prediction
45-
from synalinks.src.backend.pydantic.base import Relation
46-
from synalinks.src.backend.pydantic.base import Relations
47-
from synalinks.src.backend.pydantic.base import SimilaritySearch
48-
from synalinks.src.backend.pydantic.base import TripletSearch
49-
from synalinks.src.backend.pydantic.base import is_chat_message
50-
from synalinks.src.backend.pydantic.base import is_chat_messages
51-
from synalinks.src.backend.pydantic.base import is_embedded_entity
52-
from synalinks.src.backend.pydantic.base import is_embedding
53-
from synalinks.src.backend.pydantic.base import is_embeddings
54-
from synalinks.src.backend.pydantic.base import is_entities
55-
from synalinks.src.backend.pydantic.base import is_entity
56-
from synalinks.src.backend.pydantic.base import is_instructions
57-
from synalinks.src.backend.pydantic.base import is_knowledge_graph
58-
from synalinks.src.backend.pydantic.base import is_prediction
59-
from synalinks.src.backend.pydantic.base import is_relation
60-
from synalinks.src.backend.pydantic.base import is_relations
61-
from synalinks.src.backend.pydantic.base import is_similarity_search
62-
from synalinks.src.backend.pydantic.base import is_triplet_search
63-
from synalinks.src.embedding_models.embedding_model import EmbeddingModel
64-
from synalinks.src.initializers.initializer import Initializer
65-
from synalinks.src.knowledge_bases.knowledge_base import KnowledgeBase
66-
from synalinks.src.language_models.language_model import LanguageModel
67-
from synalinks.src.metrics.metric import Metric
68-
from synalinks.src.modules.agents.react_agent import ReACTAgent
69-
from synalinks.src.modules.core.action import Action
70-
from synalinks.src.modules.core.branch import Branch
71-
from synalinks.src.modules.core.decision import Decision
72-
from synalinks.src.modules.core.generator import Generator
73-
from synalinks.src.modules.core.generator import chat_prompt_template
74-
from synalinks.src.modules.core.generator import default_prompt_template
75-
from synalinks.src.modules.core.identity import Identity
76-
from synalinks.src.modules.core.input_module import Input
77-
from synalinks.src.modules.core.not_module import Not
78-
from synalinks.src.modules.knowledge.embedding import Embedding
79-
from synalinks.src.modules.knowledge.entity_retriever import EntityRetriever
80-
from synalinks.src.modules.knowledge.knowledge_retriever import KnowledgeRetriever
81-
from synalinks.src.modules.knowledge.update_knowledge import UpdateKnowledge
82-
from synalinks.src.modules.merging.concat import Concat
83-
from synalinks.src.modules.merging.logical_and import And
84-
from synalinks.src.modules.merging.logical_or import Or
85-
from synalinks.src.modules.merging.logical_xor import Xor
86-
from synalinks.src.modules.module import Module
87-
from synalinks.src.modules.ttc.chain_of_thought import ChainOfThought
88-
from synalinks.src.modules.ttc.self_critique import SelfCritique
89-
from synalinks.src.ops.function import Function
90-
from synalinks.src.ops.operation import Operation
91-
from synalinks.src.programs.program import Program
92-
from synalinks.src.programs.sequential import Sequential
93-
from synalinks.src.rewards.reward import Reward
7+
from synalinks.api import backend as backend
8+
from synalinks.api import callbacks as callbacks
9+
from synalinks.api import config as config
10+
from synalinks.api import datasets as datasets
11+
from synalinks.api import embedding_models as embedding_models
12+
from synalinks.api import initializers as initializers
13+
from synalinks.api import language_models as language_models
14+
from synalinks.api import metrics as metrics
15+
from synalinks.api import modules as modules
16+
from synalinks.api import ops as ops
17+
from synalinks.api import optimizers as optimizers
18+
from synalinks.api import programs as programs
19+
from synalinks.api import rewards as rewards
20+
from synalinks.api import saving as saving
21+
from synalinks.api import tree as tree
22+
from synalinks.api import utils as utils
23+
from synalinks.src.backend import DataModel as DataModel
24+
from synalinks.src.backend import name_scope as name_scope
25+
from synalinks.src.backend.common.global_state import clear_session as clear_session
26+
from synalinks.src.backend.common.json_data_model import JsonDataModel as JsonDataModel
27+
from synalinks.src.backend.common.stateless_scope import StatelessScope as StatelessScope
28+
from synalinks.src.backend.common.symbolic_data_model import (
29+
SymbolicDataModel as SymbolicDataModel,
30+
)
31+
from synalinks.src.backend.common.symbolic_scope import SymbolicScope as SymbolicScope
32+
from synalinks.src.backend.config import disable_telemetry as disable_telemetry
33+
from synalinks.src.backend.config import synalinks_home as synalinks_home
34+
from synalinks.src.backend.pydantic.base import ChatMessage as ChatMessage
35+
from synalinks.src.backend.pydantic.base import ChatMessages as ChatMessages
36+
from synalinks.src.backend.pydantic.base import ChatRole as ChatRole
37+
from synalinks.src.backend.pydantic.base import EmbeddedEntity as EmbeddedEntity
38+
from synalinks.src.backend.pydantic.base import Embeddings as Embeddings
39+
from synalinks.src.backend.pydantic.base import Entities as Entities
40+
from synalinks.src.backend.pydantic.base import Entity as Entity
41+
from synalinks.src.backend.pydantic.base import GenericInputs as GenericInputs
42+
from synalinks.src.backend.pydantic.base import GenericIO as GenericIO
43+
from synalinks.src.backend.pydantic.base import GenericOutputs as GenericOutputs
44+
from synalinks.src.backend.pydantic.base import GenericResult as GenericResult
45+
from synalinks.src.backend.pydantic.base import Instructions as Instructions
46+
from synalinks.src.backend.pydantic.base import KnowledgeGraph as KnowledgeGraph
47+
from synalinks.src.backend.pydantic.base import Prediction as Prediction
48+
from synalinks.src.backend.pydantic.base import Relation as Relation
49+
from synalinks.src.backend.pydantic.base import Relations as Relations
50+
from synalinks.src.backend.pydantic.base import SimilaritySearch as SimilaritySearch
51+
from synalinks.src.backend.pydantic.base import TripletSearch as TripletSearch
52+
from synalinks.src.backend.pydantic.base import is_chat_message as is_chat_message
53+
from synalinks.src.backend.pydantic.base import is_chat_messages as is_chat_messages
54+
from synalinks.src.backend.pydantic.base import is_embedded_entity as is_embedded_entity
55+
from synalinks.src.backend.pydantic.base import is_embedding as is_embedding
56+
from synalinks.src.backend.pydantic.base import is_embeddings as is_embeddings
57+
from synalinks.src.backend.pydantic.base import is_entities as is_entities
58+
from synalinks.src.backend.pydantic.base import is_entity as is_entity
59+
from synalinks.src.backend.pydantic.base import is_instructions as is_instructions
60+
from synalinks.src.backend.pydantic.base import is_knowledge_graph as is_knowledge_graph
61+
from synalinks.src.backend.pydantic.base import is_prediction as is_prediction
62+
from synalinks.src.backend.pydantic.base import is_relation as is_relation
63+
from synalinks.src.backend.pydantic.base import is_relations as is_relations
64+
from synalinks.src.backend.pydantic.base import (
65+
is_similarity_search as is_similarity_search,
66+
)
67+
from synalinks.src.backend.pydantic.base import is_triplet_search as is_triplet_search
68+
from synalinks.src.embedding_models.embedding_model import (
69+
EmbeddingModel as EmbeddingModel,
70+
)
71+
from synalinks.src.initializers.initializer import Initializer as Initializer
72+
from synalinks.src.knowledge_bases.knowledge_base import KnowledgeBase as KnowledgeBase
73+
from synalinks.src.language_models.language_model import LanguageModel as LanguageModel
74+
from synalinks.src.metrics.metric import Metric as Metric
75+
from synalinks.src.modules.agents.react_agent import ReACTAgent as ReACTAgent
76+
from synalinks.src.modules.core.action import Action as Action
77+
from synalinks.src.modules.core.branch import Branch as Branch
78+
from synalinks.src.modules.core.decision import Decision as Decision
79+
from synalinks.src.modules.core.generator import Generator as Generator
80+
from synalinks.src.modules.core.generator import (
81+
chat_prompt_template as chat_prompt_template,
82+
)
83+
from synalinks.src.modules.core.generator import (
84+
default_prompt_template as default_prompt_template,
85+
)
86+
from synalinks.src.modules.core.identity import Identity as Identity
87+
from synalinks.src.modules.core.input_module import Input as Input
88+
from synalinks.src.modules.core.not_module import Not as Not
89+
from synalinks.src.modules.knowledge.embedding import Embedding as Embedding
90+
from synalinks.src.modules.knowledge.entity_retriever import (
91+
EntityRetriever as EntityRetriever,
92+
)
93+
from synalinks.src.modules.knowledge.knowledge_retriever import (
94+
KnowledgeRetriever as KnowledgeRetriever,
95+
)
96+
from synalinks.src.modules.knowledge.update_knowledge import (
97+
UpdateKnowledge as UpdateKnowledge,
98+
)
99+
from synalinks.src.modules.merging.concat import Concat as Concat
100+
from synalinks.src.modules.merging.logical_and import And as And
101+
from synalinks.src.modules.merging.logical_or import Or as Or
102+
from synalinks.src.modules.merging.logical_xor import Xor as Xor
103+
from synalinks.src.modules.module import Module as Module
104+
from synalinks.src.modules.ttc.chain_of_thought import ChainOfThought as ChainOfThought
105+
from synalinks.src.modules.ttc.self_critique import SelfCritique as SelfCritique
106+
from synalinks.src.ops.function import Function as Function
107+
from synalinks.src.ops.operation import Operation as Operation
108+
from synalinks.src.programs.program import Program as Program
109+
from synalinks.src.programs.sequential import Sequential as Sequential
110+
from synalinks.src.rewards.reward import Reward as Reward
94111
from synalinks.src.version import __version__
95-
from synalinks.src.version import version
112+
from synalinks.src.version import version as version

‎synalinks/api/backend/__init__.py‎

Lines changed: 54 additions & 47 deletions
Original file line numberDiff line numberDiff line change
@@ -4,50 +4,57 @@
44
since your modifications would be overwritten.
55
"""
66

7-
from synalinks.src.backend import DataModel
8-
from synalinks.src.backend.common.global_state import clear_session
9-
from synalinks.src.backend.common.json_data_model import is_json_data_model
10-
from synalinks.src.backend.common.symbolic_data_model import is_symbolic_data_model
11-
from synalinks.src.backend.config import api_key
12-
from synalinks.src.backend.config import backend
13-
from synalinks.src.backend.config import epsilon
14-
from synalinks.src.backend.config import floatx
15-
from synalinks.src.backend.config import set_api_key
16-
from synalinks.src.backend.config import set_backend
17-
from synalinks.src.backend.config import set_epsilon
18-
from synalinks.src.backend.config import set_floatx
19-
from synalinks.src.backend.pydantic.base import ChatMessage
20-
from synalinks.src.backend.pydantic.base import ChatMessages
21-
from synalinks.src.backend.pydantic.base import ChatRole
22-
from synalinks.src.backend.pydantic.base import EmbeddedEntity
23-
from synalinks.src.backend.pydantic.base import Embedding
24-
from synalinks.src.backend.pydantic.base import Embeddings
25-
from synalinks.src.backend.pydantic.base import Entities
26-
from synalinks.src.backend.pydantic.base import Entity
27-
from synalinks.src.backend.pydantic.base import GenericInputs
28-
from synalinks.src.backend.pydantic.base import GenericIO
29-
from synalinks.src.backend.pydantic.base import GenericOutputs
30-
from synalinks.src.backend.pydantic.base import GenericResult
31-
from synalinks.src.backend.pydantic.base import Instructions
32-
from synalinks.src.backend.pydantic.base import KnowledgeGraph
33-
from synalinks.src.backend.pydantic.base import Prediction
34-
from synalinks.src.backend.pydantic.base import Relation
35-
from synalinks.src.backend.pydantic.base import Relations
36-
from synalinks.src.backend.pydantic.base import SimilaritySearch
37-
from synalinks.src.backend.pydantic.base import TripletSearch
38-
from synalinks.src.backend.pydantic.base import is_chat_message
39-
from synalinks.src.backend.pydantic.base import is_chat_messages
40-
from synalinks.src.backend.pydantic.base import is_embedded_entity
41-
from synalinks.src.backend.pydantic.base import is_embedding
42-
from synalinks.src.backend.pydantic.base import is_embeddings
43-
from synalinks.src.backend.pydantic.base import is_entities
44-
from synalinks.src.backend.pydantic.base import is_entity
45-
from synalinks.src.backend.pydantic.base import is_instructions
46-
from synalinks.src.backend.pydantic.base import is_knowledge_graph
47-
from synalinks.src.backend.pydantic.base import is_prediction
48-
from synalinks.src.backend.pydantic.base import is_relation
49-
from synalinks.src.backend.pydantic.base import is_relations
50-
from synalinks.src.backend.pydantic.base import is_similarity_search
51-
from synalinks.src.backend.pydantic.base import is_triplet_search
52-
from synalinks.src.backend.pydantic.core import is_meta_class
53-
from synalinks.src.utils.naming import get_uid
7+
from synalinks.src.backend import DataModel as DataModel
8+
from synalinks.src.backend.common.global_state import clear_session as clear_session
9+
from synalinks.src.backend.common.json_data_model import (
10+
is_json_data_model as is_json_data_model,
11+
)
12+
from synalinks.src.backend.common.symbolic_data_model import (
13+
is_symbolic_data_model as is_symbolic_data_model,
14+
)
15+
from synalinks.src.backend.config import api_key as api_key
16+
from synalinks.src.backend.config import backend as backend
17+
from synalinks.src.backend.config import epsilon as epsilon
18+
from synalinks.src.backend.config import floatx as floatx
19+
from synalinks.src.backend.config import is_telemetry_enabled as is_telemetry_enabled
20+
from synalinks.src.backend.config import set_api_key as set_api_key
21+
from synalinks.src.backend.config import set_backend as set_backend
22+
from synalinks.src.backend.config import set_epsilon as set_epsilon
23+
from synalinks.src.backend.config import set_floatx as set_floatx
24+
from synalinks.src.backend.pydantic.base import ChatMessage as ChatMessage
25+
from synalinks.src.backend.pydantic.base import ChatMessages as ChatMessages
26+
from synalinks.src.backend.pydantic.base import ChatRole as ChatRole
27+
from synalinks.src.backend.pydantic.base import EmbeddedEntity as EmbeddedEntity
28+
from synalinks.src.backend.pydantic.base import Embedding as Embedding
29+
from synalinks.src.backend.pydantic.base import Embeddings as Embeddings
30+
from synalinks.src.backend.pydantic.base import Entities as Entities
31+
from synalinks.src.backend.pydantic.base import Entity as Entity
32+
from synalinks.src.backend.pydantic.base import GenericInputs as GenericInputs
33+
from synalinks.src.backend.pydantic.base import GenericIO as GenericIO
34+
from synalinks.src.backend.pydantic.base import GenericOutputs as GenericOutputs
35+
from synalinks.src.backend.pydantic.base import GenericResult as GenericResult
36+
from synalinks.src.backend.pydantic.base import Instructions as Instructions
37+
from synalinks.src.backend.pydantic.base import KnowledgeGraph as KnowledgeGraph
38+
from synalinks.src.backend.pydantic.base import Prediction as Prediction
39+
from synalinks.src.backend.pydantic.base import Relation as Relation
40+
from synalinks.src.backend.pydantic.base import Relations as Relations
41+
from synalinks.src.backend.pydantic.base import SimilaritySearch as SimilaritySearch
42+
from synalinks.src.backend.pydantic.base import TripletSearch as TripletSearch
43+
from synalinks.src.backend.pydantic.base import is_chat_message as is_chat_message
44+
from synalinks.src.backend.pydantic.base import is_chat_messages as is_chat_messages
45+
from synalinks.src.backend.pydantic.base import is_embedded_entity as is_embedded_entity
46+
from synalinks.src.backend.pydantic.base import is_embedding as is_embedding
47+
from synalinks.src.backend.pydantic.base import is_embeddings as is_embeddings
48+
from synalinks.src.backend.pydantic.base import is_entities as is_entities
49+
from synalinks.src.backend.pydantic.base import is_entity as is_entity
50+
from synalinks.src.backend.pydantic.base import is_instructions as is_instructions
51+
from synalinks.src.backend.pydantic.base import is_knowledge_graph as is_knowledge_graph
52+
from synalinks.src.backend.pydantic.base import is_prediction as is_prediction
53+
from synalinks.src.backend.pydantic.base import is_relation as is_relation
54+
from synalinks.src.backend.pydantic.base import is_relations as is_relations
55+
from synalinks.src.backend.pydantic.base import (
56+
is_similarity_search as is_similarity_search,
57+
)
58+
from synalinks.src.backend.pydantic.base import is_triplet_search as is_triplet_search
59+
from synalinks.src.backend.pydantic.core import is_meta_class as is_meta_class
60+
from synalinks.src.utils.naming import get_uid as get_uid

‎synalinks/api/callbacks/__init__.py‎

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,11 @@
44
since your modifications would be overwritten.
55
"""
66

7-
from synalinks.src.callbacks.callback import Callback
8-
from synalinks.src.callbacks.callback_list import CallbackList
9-
from synalinks.src.callbacks.csv_logger import CSVLogger
10-
from synalinks.src.callbacks.history import History
11-
from synalinks.src.callbacks.progbar_logger import ProgbarLogger
12-
from synalinks.src.callbacks.program_checkpoint import ProgramCheckpoint
7+
from synalinks.src.callbacks.callback import Callback as Callback
8+
from synalinks.src.callbacks.callback_list import CallbackList as CallbackList
9+
from synalinks.src.callbacks.csv_logger import CSVLogger as CSVLogger
10+
from synalinks.src.callbacks.history import History as History
11+
from synalinks.src.callbacks.progbar_logger import ProgbarLogger as ProgbarLogger
12+
from synalinks.src.callbacks.program_checkpoint import (
13+
ProgramCheckpoint as ProgramCheckpoint,
14+
)

0 commit comments

Comments
 (0)