symbolicai 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

symai/extended/graph.py

@@ -21,7 +21,7 @@ If more than one entity pair is extracted from the same sentence, then the CSV f
 
 class GraphPreProcessor(PreProcessor):
     def __call__(self, argument):
-        return f'$> {argument.args[0]!s} =>'
+        return f"$> {argument.args[0]!s} =>"
 
 
 class Graph(Expression):
@@ -29,26 +29,37 @@ class Graph(Expression):
     def static_context(self) -> str:
         return GRAPH_DESCRIPTION
 
-    def __init__(self, formatter: Callable = _DEFAULT_SENTENCE_FORMATTER, n_workers: int = 1, verbose: bool = False, **kwargs):
+    def __init__(
+        self,
+        formatter: Callable = _DEFAULT_SENTENCE_FORMATTER,
+        n_workers: int = 1,
+        verbose: bool = False,
+        **kwargs,
+    ):
         super().__init__(**kwargs)
-        self.formatter       = formatter
-        self.n_workers       = n_workers
+        self.formatter = formatter
+        self.n_workers = n_workers
         self.sym_return_type = Graph
-        self.verbose         = verbose
+        self.verbose = verbose
 
     def process_symbol(self, s, *_args, **kwargs):
-        res = ''
+        res = ""
 
-        @core.few_shot(prompt="Extract relationships between entities:\n",
-                  examples=Prompt([
-                        '$> John has a dog. =>John, dog, 1 EOF',
-                        '$> Karl has two sons. =>Karl, sons, 2 EOF',
-                        '$> Similarly, the term general linguistics is used to distinguish core linguistics from other types of study =>general linguistics, core linguistics, 1 EOF',
-                        '$> X has Y and Z has Y =>X, Y, 1\nZ, Y, 1 EOF',
-                  ]),
-                  pre_processors=[GraphPreProcessor()],
-                  post_processors=[StripPostProcessor()],
-                  stop=['EOF'], **kwargs)
+        @core.few_shot(
+            prompt="Extract relationships between entities:\n",
+            examples=Prompt(
+                [
+                    "$> John has a dog. =>John, dog, 1 EOF",
+                    "$> Karl has two sons. =>Karl, sons, 2 EOF",
+                    "$> Similarly, the term general linguistics is used to distinguish core linguistics from other types of study =>general linguistics, core linguistics, 1 EOF",
+                    "$> X has Y and Z has Y =>X, Y, 1\nZ, Y, 1 EOF",
+                ]
+            ),
+            pre_processors=[GraphPreProcessor()],
+            post_processors=[StripPostProcessor()],
+            stop=["EOF"],
+            **kwargs,
+        )
         def _func(_, text) -> str:
             pass
 
@@ -57,17 +68,19 @@ class Graph(Expression):
                 UserMessage(str(s))
             r = _func(self, s)
             rec = str(r)
-            lines = rec.split('\n')
+            lines = rec.split("\n")
             for line in lines:
                 stripped_line = line.strip()
                 if len(stripped_line) > 0:
-                    csv = stripped_line.split(',')
+                    csv = stripped_line.split(",")
                     try:
-                        if len(csv) == 3 and \
-                            csv[0].strip() != '' and \
-                                csv[1].strip() != '' and \
-                                    int(csv[2].strip()) > 0:
-                            res += stripped_line + '\n'
+                        if (
+                            len(csv) == 3
+                            and csv[0].strip() != ""
+                            and csv[1].strip() != ""
+                            and int(csv[2].strip()) > 0
+                        ):
+                            res += stripped_line + "\n"
                     except Exception as e:
                         if self.verbose:
                             UserMessage(str(e))
@@ -75,7 +88,7 @@ class Graph(Expression):
         return res
 
     def forward(self, sym: Symbol, **_kwargs) -> Symbol:
-        res = 'source,target,value\n'
+        res = "source,target,value\n"
         sym_list = self.formatter(sym).value
         if self.n_workers == 1:
             for s in sym_list:
@@ -87,4 +100,5 @@ class Graph(Expression):
             res += r
         return res
 
+
 _DEFAULT_SENTENCE_FORMATTER = SentenceFormatter()

symai/extended/html_style_template.py

@@ -61,13 +61,15 @@ class HtmlStyleTemplate(Expression):
         super().__init__(**kwargs)
         self.html_template_seq = Template()
         self.html_template_seq.template_ = HEADER_STYLE_DESCRIPTION
-        self.html_stream = Stream(
-            self.html_template_seq
+        self.html_stream = Stream(self.html_template_seq)
+        self.style_template = Style(
+            description=HTML_STREAM_STYLE_DESCRIPTION,
+            libraries=[
+                "https://cdn.jsdelivr.net/npm/bootstrap@5.2.3/dist/css/bootstrap.min.css",
+                "https://cdn.jsdelivr.net/npm/bootstrap@5.2.3/dist/js/bootstrap.bundle.min.js",
+                "https://ajax.googleapis.com/ajax/libs/jquery/3.6.1/jquery.min.js",
+            ],
         )
-        self.style_template = Style(description=HTML_STREAM_STYLE_DESCRIPTION,
-                                       libraries=['https://cdn.jsdelivr.net/npm/bootstrap@5.2.3/dist/css/bootstrap.min.css',
-                                                  'https://cdn.jsdelivr.net/npm/bootstrap@5.2.3/dist/js/bootstrap.bundle.min.js',
-                                                  'https://ajax.googleapis.com/ajax/libs/jquery/3.6.1/jquery.min.js'])
 
     def forward(self, sym: Symbol, **kwargs) -> Symbol:
         """The `render` method takes a `Symbol` as an argument and returns a `Symbol` containing the rendered news.
@@ -79,8 +81,12 @@ class HtmlStyleTemplate(Expression):
         if not isinstance(sym, Symbol):
             sym = Symbol(sym)
         html_data = list(self.html_stream(sym, **kwargs))
-        style_data = [str(self.style_template(html,
-                                              template=HTML_TEMPLATE_STYLE,
-                                              placeholder='{{placeholder}}',
-                                              **kwargs)) for html in html_data]
-        return Symbol('\n'.join(style_data))
+        style_data = [
+            str(
+                self.style_template(
+                    html, template=HTML_TEMPLATE_STYLE, placeholder="{{placeholder}}", **kwargs
+                )
+            )
+            for html in html_data
+        ]
+        return Symbol("\n".join(style_data))

symai/extended/interfaces/blip_2.py

@@ -1,4 +1,3 @@
-
 from ... import core
 from ...symbol import Expression
 
@@ -12,4 +11,5 @@ class blip_2(Expression):
         @core.caption(image=image, prompt=query, **kwargs)
         def _func(_) -> str:
             pass
+
         return self.sym_return_type(_func(self))

symai/extended/interfaces/clip.py

@@ -1,4 +1,3 @@
-
 import numpy as np
 
 from ... import core
@@ -10,8 +9,11 @@ class clip(Expression):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
 
-    def __call__(self, image: str | bytes | None = None, text: list[str] | None = None, **kwargs) -> "clip":
+    def __call__(
+        self, image: str | bytes | None = None, text: list[str] | None = None, **kwargs
+    ) -> "clip":
         @core.text_vision(image=image, text=text, **kwargs)
         def _func(_) -> np.ndarray:
             pass
+
         return self.sym_return_type(_func(self))

symai/extended/interfaces/console.py

@@ -10,12 +10,14 @@ class console(Expression):
 
     def __call__(self, *args, **kwargs) -> "console":
         def _handler(payload):
-            args_ = payload.get('args', ())
-            message = ' '.join(str(arg) for arg in args_)
+            args_ = payload.get("args", ())
+            message = " ".join(str(arg) for arg in args_)
             UserMessage(message)
 
-        kwargs['handler'] = _handler
+        kwargs["handler"] = _handler
+
         @core.output(**kwargs)
         def _func(_, *args):
             pass
+
         return self.sym_return_type(_func(self, *args))

symai/extended/interfaces/dall_e.py

@@ -8,9 +8,11 @@ class dall_e(Expression):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
 
-    def __call__(self, sym: Symbol, operation: str = 'create', **kwargs) -> GPTImageResult:
+    def __call__(self, sym: Symbol, operation: str = "create", **kwargs) -> GPTImageResult:
         sym = self._to_symbol(sym)
+
         @core.draw(operation=operation, **kwargs)
         def _func(_) -> GPTImageResult:
             pass
+
         return _func(sym)

symai/extended/interfaces/file.py

@@ -9,7 +9,9 @@ class file(Expression):
 
     def __call__(self, path: Symbol, **kwargs) -> "file":
         path = self._to_symbol(path)
+
         @core.opening(path=path.value, **kwargs)
         def _func(_) -> str:
             pass
+
         return self.sym_return_type(_func(self))

symai/extended/interfaces/flux.py

@@ -8,9 +8,11 @@ class flux(Expression):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
 
-    def __call__(self, prompt: str, operation: str = 'create', **kwargs) -> FluxResult:
+    def __call__(self, prompt: str, operation: str = "create", **kwargs) -> FluxResult:
         prompt = self._to_symbol(prompt)
+
         @core.draw(operation=operation, **kwargs)
         def _func(_) -> FluxResult:
             pass
+
         return _func(prompt)

symai/extended/interfaces/gpt_image.py

@@ -8,9 +8,18 @@ class gpt_image(Expression):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
 
-    def __call__(self, prompt: str | None = None, operation: str = 'create', engine: str | None = None, model: str | None = None, **kwargs) -> GPTImageResult:
-            prompt = self._to_symbol(prompt)
-            @core.draw(engine=engine or "drawing", operation=operation, model=model, **kwargs)
-            def _func(_) -> GPTImageResult:
-                pass
-            return _func(prompt)
+    def __call__(
+        self,
+        prompt: str | None = None,
+        operation: str = "create",
+        engine: str | None = None,
+        model: str | None = None,
+        **kwargs,
+    ) -> GPTImageResult:
+        prompt = self._to_symbol(prompt)
+
+        @core.draw(engine=engine or "drawing", operation=operation, model=model, **kwargs)
+        def _func(_) -> GPTImageResult:
+            pass
+
+        return _func(prompt)

symai/extended/interfaces/input.py

@@ -2,7 +2,7 @@ from ... import core
 from ...symbol import Expression
 
 
-class input(Expression): # noqa
+class input(Expression):  # noqa
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
@@ -11,4 +11,5 @@ class input(Expression): # noqa
         @core.userinput(**kwargs)
         def _func(_, message) -> str:
             pass
+
         return self.sym_return_type(_func(self, message))

symai/extended/interfaces/llava.py

@@ -1,4 +1,3 @@
-
 from ... import core
 from ...symbol import Expression
 
@@ -12,4 +11,5 @@ class llava(Expression):
         @core.caption(image=image, prompt=query, **kwargs)
         def _func(_) -> str:
             pass
+
         return self.sym_return_type(_func(self))

symai/extended/interfaces/{naive_webscraping.py → naive_scrape.py}

@@ -1,9 +1,9 @@
 from ... import core
-from ...backend.engines.webscraping.engine_requests import RequestsResult
+from ...backend.engines.scrape.engine_requests import RequestsResult
 from ...symbol import Expression
 
 
-class naive_webscraping(Expression):
+class naive_scrape(Expression):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
@@ -14,4 +14,5 @@ class naive_webscraping(Expression):
             # The fallback path may inject debugging kwargs like `error`/`stack_trace`;
             # accept and ignore them so EngineRepository can surface structured failures.
             return None
+
         return _func(self)

symai/extended/interfaces/naive_vectordb.py

@@ -4,7 +4,7 @@ from ...utils import UserMessage
 
 
 class naive_vectordb(Expression):
-    def __init__(self, index_name = VectorDBIndexEngine._default_index_name, *args, **kwargs):
+    def __init__(self, index_name=VectorDBIndexEngine._default_index_name, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.index_name = index_name
         self.name = self.__class__.__name__
@@ -15,7 +15,7 @@ class naive_vectordb(Expression):
         operation: str = "search",
         index_name: str | None = None,
         storage_file: str | None = None,
-        **kwargs
+        **kwargs,
     ) -> VectorDBResult:
         index = self.index_name if index_name is None else index_name
         if operation == "search":

symai/extended/interfaces/ocr.py

@@ -9,9 +9,11 @@ class ocr(Expression):
         self.name = self.__class__.__name__
 
     def __call__(self, image_url: str, **kwargs) -> ApiLayerResult:
-        if not image_url.startswith('http'):
-            image_url = f'file://{image_url}'
+        if not image_url.startswith("http"):
+            image_url = f"file://{image_url}"
+
         @core.ocr(image=image_url, **kwargs)
         def _func(_) -> ApiLayerResult:
             pass
+
         return _func(self)

symai/extended/interfaces/openai_search.py

@@ -10,7 +10,9 @@ class openai_search(Expression):
 
     def __call__(self, query: Symbol, **kwargs) -> SearchResult:
         query = self._to_symbol(query)
+
         @core.search(query=query.value, **kwargs)
         def _func(_) -> SearchResult:
             pass
+
         return _func(self)

symai/extended/interfaces/parallel.py

@@ -0,0 +1,30 @@
+from ... import core
+from ...backend.engines.search.engine_parallel import ExtractResult, SearchResult
+from ...symbol import Expression, Symbol
+
+
+class parallel(Expression):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.name = self.__class__.__name__
+
+    def search(self, query: Symbol, **kwargs) -> SearchResult:
+        query = self._to_symbol(query)
+
+        @core.search(query=query.value, **kwargs)
+        def _func(_) -> SearchResult:
+            pass
+
+        return _func(self)
+
+    def scrape(self, url: str, **kwargs) -> ExtractResult:
+        symbol = self._to_symbol(url)
+        options = dict(kwargs)
+        options.pop("query", None)
+        options["url"] = symbol.value
+
+        @core.search(query="", **options)
+        def _func(_, *_args, **_inner_kwargs) -> ExtractResult:
+            return None
+
+        return _func(self)

symai/extended/interfaces/perplexity.py

@@ -10,7 +10,9 @@ class perplexity(Expression):
 
     def __call__(self, query: Symbol, **kwargs) -> SearchResult:
         query = self._to_symbol(query)
+
         @core.search(query=query.value, **kwargs)
         def _func(_) -> SearchResult:
             pass
+
         return _func(self)

symai/extended/interfaces/pinecone.py

@@ -4,15 +4,17 @@ from ...utils import UserMessage
 
 
 class pinecone(Expression):
-    def __init__(self, index_name = PineconeIndexEngine._default_index_name, *args, **kwargs):
+    def __init__(self, index_name=PineconeIndexEngine._default_index_name, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.index_name = index_name
         self.name = self.__class__.__name__
 
-    def __call__(self, stmt: str, operation: str = "search", index_name = None, **kwargs) -> PineconeResult:
-        stmt  = self._to_symbol(stmt)
+    def __call__(
+        self, stmt: str, operation: str = "search", index_name=None, **kwargs
+    ) -> PineconeResult:
+        stmt = self._to_symbol(stmt)
         index = self.index_name if index_name is None else index_name
-        if   operation == "search":
+        if operation == "search":
             return self.get(query=stmt.embedding, index_name=index, ori_query=stmt.value, **kwargs)
         if operation == "add":
             return self.add(doc=stmt.zip(), index_name=index, **kwargs)

symai/extended/interfaces/python.py

@@ -9,7 +9,9 @@ class python(Expression):
 
     def __call__(self, expr: str, **kwargs) -> Result:
         sym = self._to_symbol(expr)
+
         @core.execute(**kwargs)
         def _func(_) -> Result:
             pass
+
         return _func(sym)

symai/extended/interfaces/serpapi.py

@@ -10,7 +10,9 @@ class serpapi(Expression):
 
     def __call__(self, query: Symbol, **kwargs) -> SearchResult:
         query = self._to_symbol(query)
+
         @core.search(query=query.value, **kwargs)
         def _func(_) -> SearchResult:
             pass
+
         return _func(self)

symai/extended/interfaces/terminal.py

@@ -1,4 +1,3 @@
-
 from ...shellsv import process_command
 from ...symbol import Expression
 

symai/extended/interfaces/tts.py

@@ -7,8 +7,9 @@ class tts(Expression):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
 
-    def __call__(self, prompt: Symbol, path: str, voice: str = 'nova', **kwargs) -> Result:
+    def __call__(self, prompt: Symbol, path: str, voice: str = "nova", **kwargs) -> Result:
         @core.text_to_speech(prompt=str(prompt), path=path, voice=voice, **kwargs)
         def _func(_) -> Result:
             pass
+
         return _func(self)

symai/extended/interfaces/whisper.py

@@ -8,8 +8,9 @@ class whisper(Expression):
         super().__init__(*args, **kwargs)
         self.name = self.__class__.__name__
 
-    def __call__(self, audio_path: str, operation: str = 'decode', **kwargs) -> WhisperResult:
+    def __call__(self, audio_path: str, operation: str = "decode", **kwargs) -> WhisperResult:
         @core.speech_to_text(audio=audio_path, prompt=operation, **kwargs)
         def _func(_) -> WhisperResult:
             pass
+
         return _func(self)

symai/extended/interfaces/wolframalpha.py

@@ -12,4 +12,5 @@ class wolframalpha(Expression):
         @core.expression(**kwargs)
         def _func(_, expr: str) -> WolframResult:
             pass
+
         return _func(self, expr)

symai/extended/metrics/__init__.py

@@ -1,6 +1,6 @@
 from . import similarity as _similarity
 
-__all__ = getattr(_similarity, "__all__", None) # noqa
+__all__ = getattr(_similarity, "__all__", None)  # noqa
 if __all__ is None:
     __all__ = [name for name in dir(_similarity) if not name.startswith("_")]
 

symai/extended/metrics/similarity.py

@@ -1,6 +1,7 @@
-'''
+"""
 @TODO: This must go away. As soon as we have Qdrant, everything related to vectordb must be cleaned up.
-'''
+"""
+
 import random
 
 import numpy as np
@@ -60,6 +61,7 @@ def derridaean_similarity(vectors, query_vector):
     :param query_vector: vector
     :return: derridaean similarity between the vector and the matrix of vectors
     """
+
     def random_change(value):
         return value + random.uniform(-0.2, 0.2)
 
@@ -74,6 +76,7 @@ def adams_similarity(vectors, query_vector):
     :param query_vector: vector
     :return: adams similarity between the vector and the matrix of vectors
     """
+
     def adams_change(_value):
         return 0.42
 

symai/extended/os_command.py

@@ -61,43 +61,52 @@ Write an executable command that starts a process according to the user query, p
 
 
 class OSCommand(Expression):
-    def __init__(self, programs:        list[str],
-                       metadata:   dict[str, str] | None = None,
-                       verbose:              bool = False,
-                       os_platform:           str = 'auto',
-                       **kwargs):
+    def __init__(
+        self,
+        programs: list[str],
+        metadata: dict[str, str] | None = None,
+        verbose: bool = False,
+        os_platform: str = "auto",
+        **kwargs,
+    ):
         if metadata is None:
             metadata = {}
         super().__init__(**kwargs)
-        self.verbose:            bool = verbose
-        self.os_platform:         str = os_platform
-        self.programs:      list[str] = programs
-        self.meta: dict[str, str]     = metadata
+        self.verbose: bool = verbose
+        self.os_platform: str = os_platform
+        self.programs: list[str] = programs
+        self.meta: dict[str, str] = metadata
 
-        if self.os_platform == 'auto':
+        if self.os_platform == "auto":
             self.os_platform = platform.platform()
         if len(programs) == 0:
-            UserMessage('No programs specified!', raise_with=Exception)
+            UserMessage("No programs specified!", raise_with=Exception)
 
     def execute_os_command(self, *args, **_kwargs):
         command = args[0]
-        UserMessage(f'Executing {self.os_platform} command: {command}')
-        if 'linux' in self.os_platform.lower():
+        UserMessage(f"Executing {self.os_platform} command: {command}")
+        if "linux" in self.os_platform.lower():
             return [subprocess.run(["bash", "-c", str(command)], check=False)]
-        if 'windows' in self.os_platform.lower():
+        if "windows" in self.os_platform.lower():
             return [subprocess.run(["powershell", "-Command", str(command)], check=False)]
-        if 'mac' in self.os_platform.lower():
+        if "mac" in self.os_platform.lower():
             return [subprocess.run(["bash", "-c", str(command)], check=False)]
-        UserMessage('Unsupported platform!', raise_with=Exception)
+        UserMessage("Unsupported platform!", raise_with=Exception)
         return []
 
     def forward(self, sym: Symbol, **kwargs) -> Expression:
         sym = self._to_symbol(sym)
-        kwargs['verbose'] = self.verbose
+        kwargs["verbose"] = self.verbose
 
-        prompt = Context.format(programs=self.programs,
-                                platform=self.os_platform,
-                                query=sym,
-                                metadata=self.meta)
+        prompt = Context.format(
+            programs=self.programs, platform=self.os_platform, query=sym, metadata=self.meta
+        )
         command = sym.query(prompt, post_processors=[CodeExtractPostProcessor()], **kwargs)
-        return self.sym_return_type(self.output(expr=self.execute_os_command, raw_input=True, processed_input=command.value, **kwargs))
+        return self.sym_return_type(
+            self.output(
+                expr=self.execute_os_command,
+                raw_input=True,
+                processed_input=command.value,
+                **kwargs,
+            )
+        )