versionhq 1.2.4.5__py3-none-any.whl → 1.2.4.6__py3-none-any.whl

versionhq/tool/gpt/file_search.py

@@ -0,0 +1,163 @@
+from typing import List, Dict, Any, Optional, Tuple
+
+from versionhq.tool.gpt import openai_client
+from versionhq.tool.gpt._enum import GPTFilterTypeEnum
+from versionhq._utils import is_valid_enum, UsageMetrics, ErrorType
+
+
+def is_valid_vector_store_id(id: str | list[str]) -> bool:
+    if isinstance(id, list):
+        for item in id:
+            if not id.startswith("vs_"):
+                return False
+    else:
+        return id.startswith("vs_")
+
+
+class FilterSchema:
+    class Filter:
+        type: GPTFilterTypeEnum = GPTFilterTypeEnum.eq
+        property: str = None
+        value: str = None
+
+        def __init__(self, **kwargs):
+            for k, v in kwargs:
+                if hasattr(self, k):
+                    setattr(self, k, v)
+
+        def _convert_to_schema(self) -> Dict[str, str] | None:
+            return { "type": self.type, "property": self.property, "value": self.value } if self.property and self.value else None
+
+    logic_type: str = "and"  # or
+    filters: List[Filter] = list()
+    filter_params: Dict[str, Any] = None
+
+    def __init__(self, logic_type: str = None, filters: List[Filter] | Filter = None, filter_params: Dict[str, Any] = None, **kwargs):
+        if logic_type == "and" | "or":
+            self.logic_type = logic_type
+
+        if filter_params:
+            filter = FilterSchema.Filter()
+            for k, v in filter_params.items():
+                if k in FilterSchema.Filter.__dict__.keys():
+                    if k == "type" and is_valid_enum(val=v, enum=GPTFilterTypeEnum):
+                        setattr(filter, k, v if isinstance(v, str) else v.value)
+                    else:
+                        setattr(filter, k, v)
+            self.filters.append(filter)
+
+        if filters:
+            match filters:
+                case list():
+                    if self.filters:
+                        self.filters.extend(filters)
+                    else:
+                        self.filters = filters
+
+                case FilterSchema.Filter():
+                    if self.filters:
+                        self.filters.append(filters)
+                    else:
+                        self.filters = [filters]
+
+                case _:
+                    pass
+
+        if kwargs:
+            for k, v in kwargs:
+                if hasattr(self, k): setattr(self, k, v)
+
+    @property
+    def schema(self) -> Dict[str, Any] | None:
+        if self.type and len(self.items) > 1:
+            return {
+                "type": self.type,
+                "filters": [item._convert_to_schema() for item in self.items if isinstance(item, FilterSchema.Filter)]
+            }
+        elif self.filters:
+            return self.filters[0]._convert_to_schema()
+        else:
+            return None
+
+
+class GPTToolFileSearch:
+    model: str = "gpt-4o"
+    input: str = None
+    vector_store_ids: List[str] = list()
+    max_num_results: int = 2
+    include: List[str] = ["output[*].file_search_call.search_results"]
+    filters: Optional[FilterSchema] = None
+
+    def __init__(
+            self,
+            input: str,
+            vector_store_ids: str | List[str],
+            model: str = None,
+            max_num_results: int = None,
+            include: List[str] = None,
+            filters: FilterSchema | Dict[str, Any] = None,
+            **kwargs,
+        ):
+        if not input or not vector_store_ids:
+            return None
+
+        if not is_valid_vector_store_id(id=vector_store_ids):
+            return None
+
+        self.input = input
+        self.vector_store_ids = vector_store_ids if isinstance(vector_store_ids, list) else [vector_store_ids]
+        self.model = model if model else self.model
+        self.max_num_results = max_num_results if max_num_results else self.max_num_results
+        self.include = include if include else self.include
+        self.filters = filters if filters else None
+        if kwargs:
+            for k, v in kwargs.items():
+                if hasattr(self, k):
+                    setattr(self, k, v)
+
+
+    def run(self) -> Tuple[str, List[Dict[str, Any]], UsageMetrics] | None:
+        raw_res = ""
+        annotations = list()
+        usage = UsageMetrics()
+
+        try:
+            res = openai_client.responses.create(**self.schema)
+            if not res:
+                usage.record_errors(ErrorType.TOOL)
+            else:
+                raw_res = res.output[1].content[0].text
+                annotations = [{ "index": item.index, "file_id": item.file_id, "filename": item.filename }
+                                for item in res.output[1].content[0].annotations]
+                usage.record_token_usage(**res.usage.__dict__)
+            return raw_res, annotations, usage
+        except:
+            usage.record_errors(ErrorType.TOOL)
+            return raw_res, annotations, usage
+
+
+    @property
+    def tool_schema(self) -> Dict[str, Any]:
+        if self.filters:
+            return [
+                {
+                    "type": "file_search",
+                    "vector_store_ids": self.vector_store_ids,
+                    "max_num_results": self.max_num_results,
+                    "filters": self.filters.schema,
+                }
+            ]
+        else:
+            return [
+                {
+                    "type": "file_search",
+                    "vector_store_ids": self.vector_store_ids,
+                    "max_num_results": self.max_num_results,
+                }
+            ]
+
+
+    @property
+    def schema(self) -> Dict[str, Any]:
+        schema = dict(model=self.model, tools=self.tool_schema, input=self.input)
+        return schema

versionhq/tool/gpt/web_search.py

@@ -0,0 +1,89 @@
+from typing import Dict, Any, Optional, Tuple, List
+
+from versionhq.tool.gpt import openai_client
+from versionhq.tool.gpt._enum import GPTSizeEnum
+from versionhq._utils import is_valid_enum, UsageMetrics, ErrorType
+
+
+class GPTToolWebSearch:
+    """A class to manage Web Search tools by OpenAI."""
+
+    model: str = "gpt-4o"
+    input: str = None
+    location_type: str = None  # "approximate"
+    country: str = None  # "GB"
+    city: str = None # "London"
+    region: str = None # "London"
+    search_content_size: str = GPTSizeEnum.MEDIUM.value
+    _user_location: Optional[Dict[str, str]] = None
+
+
+    def __init__(
+            self,
+            model: str = None,
+            input: str = None,
+            location_type: str = None,
+            country: str = None,
+            city: str = None,
+            region: str = None,
+            search_content_size = str | GPTSizeEnum,
+            **kwargs,
+        ):
+        self.model = model if model else self.model
+        self.input = input if input else self.input
+        if country and city and region:
+            self.location_type = location_type if location_type else "approximate"
+            self.country = country
+            self.city = city
+            self.region = region
+            self._user_location = dict(type=self.location_type, country=self.country, city=self.city, region=self.region)
+
+        if is_valid_enum(val=search_content_size, enum=GPTSizeEnum):
+            self.search_content_size = search_content_size if isinstance(search_content_size, str) else search_content_size.value
+
+        if kwargs:
+            for k, v in kwargs.items():
+                if hasattr(self, k):
+                    setattr(self, k, v)
+
+
+    def run(self) -> Tuple[str, List[Dict[str, str]], UsageMetrics]:
+        """Runs the tool and returns text response, annotations, and usage metrics."""
+
+        raw_res = ""
+        annotations = list()
+        usage = UsageMetrics()
+
+        try:
+            res = openai_client.responses.create(**self.schema)
+            if not res:
+                usage.record_errors(ErrorType.TOOL)
+            else:
+                raw_res = res.output[1].content[0].text
+                annotations = [{ "title": item.title, "url": item.url } for item in res.output[1].content[0].annotations]
+                usage.record_token_usage(**res.usage.__dict__)
+            return raw_res, annotations, usage
+        except:
+            usage.record_errors(ErrorType.TOOL)
+            return raw_res, annotations, usage
+
+
+    @property
+    def tool_schema(self) -> Dict[str, Any]:
+        if self._user_location:
+            return {
+                "type": "web_search_preview",
+                "user_location": self._user_location,
+                "search_context_size": self.search_content_size
+            }
+        else:
+            return {
+                "type": "web_search_preview",
+                "search_context_size": self.search_content_size
+            }
+
+
+    @property
+    def schema(self) -> Dict[str, Any]:
+        schema = dict(model=self.model, tools=[self.tool_schema,], input=self.input)
+        return schema

{versionhq-1.2.4.5.dist-info → versionhq-1.2.4.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.4.5
+Version: 1.2.4.6
 Summary: Autonomous agent networks for task automation with multi-step reasoning.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.2.4.5.dist-info → versionhq-1.2.4.6.dist-info}/RECORD

@@ -1,33 +1,35 @@
-versionhq/__init__.py,sha256=8XLNCUY-3bjs8eI6mMUDBZLeY_Gj39QTut0tpzzyE3c,3026
-versionhq/_prompt/auto_feedback.py,sha256=QDYd8mKlluDaOjHIfL9B8Tr5tg40tsDVyBXqCuUdTCU,3800
+versionhq/__init__.py,sha256=3MWBL0YKctLd8VquoiMDwpdvaJODfwF1LYm4Hf84zbk,3333
+versionhq/_prompt/auto_feedback.py,sha256=bbj37yTa11lRHpx-sV_Wmpb4dVnDBB7_v8ageUobHXY,3780
 versionhq/_prompt/constants.py,sha256=DOwUFnVVObEFqgnaMCDnW8fnw1oPMgS8JAqOiTuqleI,932
-versionhq/_prompt/model.py,sha256=kokH7axDDPEZKPBmrOKi5L0aZLtApdVOJDKsMjkJmvw,8020
-versionhq/_utils/__init__.py,sha256=UggL2r-idlWDh0cIPFLyJ7AvO17NLzhjheW4IBFLBj4,300
+versionhq/_prompt/model.py,sha256=wJlDM9yzrqlXWxyw4HkYQzPii2MPfqkgTF3qhXoJN2M,8038
+versionhq/_utils/__init__.py,sha256=TOd3U_VCjvLzt0w-KV9cM1_ozEjzffhjyKX3F_JaqZg,418
+versionhq/_utils/convert_img_url.py,sha256=BlINw4RQ632m9P4FJbqzqYlzTLESBTRkhkstAopnNNY,408
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
+versionhq/_utils/is_valid_enum.py,sha256=vGGIuvhDnFU2fUyyFxJyjw-NfByK0vfFAu1ShaHBeZE,720
 versionhq/_utils/is_valid_url.py,sha256=m8Mswvb-90FJtx1Heq6hPFDbwGgrv_R3wSbZQmEPM9Q,379
-versionhq/_utils/llm_as_a_judge.py,sha256=RM0oYfoeanuUyUL3Ewl6_8Xn1F5Axd285UMH46kxG1I,2378
+versionhq/_utils/llm_as_a_judge.py,sha256=2XcuFqEVb6P6vHxWQNv8XM2b4APyvEir5QI9TPgHt_U,2353
 versionhq/_utils/logger.py,sha256=iHxGjm3BvUo5dHKLU88_pc0Z45wzSHOjyJGQkb7OADk,3255
 versionhq/_utils/process_config.py,sha256=YTGY_erW335RfceQfzS18YAqq-AAb-iSvKSjN7noD2E,782
-versionhq/_utils/usage_metrics.py,sha256=zaoH6xjWX69UrQJmViBiX3sEUpnwSoHaapCPfWU2oM8,2632
+versionhq/_utils/usage_metrics.py,sha256=gDK6fZgT1njX4iPIPFapWxfxIiz-zZYv72p0u6M055U,3387
 versionhq/_utils/vars.py,sha256=bZ5Dx_bFKlt3hi4-NNGXqdk7B23If_WaTIju2fiTyPQ,57
 versionhq/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/agent/inhouse_agents.py,sha256=D2WAiXCYsnQK3_Fe7CbbtvXsHWOaN6vde6m_QoW7fH4,2629
-versionhq/agent/model.py,sha256=auIrSWqt4NnANgA2YOCS2l9kxPCWkZG7YUD22NjrkAA,24810
+versionhq/agent/model.py,sha256=kEsJri5RWSAzQ4jNYhS3WYSJTNyTT1huqyTlqjSK62E,26969
 versionhq/agent/parser.py,sha256=riG0dkdQCxH7uJ0AbdVdg7WvL0BXhUgJht0VtQvxJBc,4082
 versionhq/agent/rpm_controller.py,sha256=grezIxyBci_lDlwAlgWFRyR5KOocXeOhYkgN02dNFNE,2360
 versionhq/agent/TEMPLATES/Backstory.py,sha256=dkfuATUQ2g2WoUKkmgAIch-RB--bektGoQaUlsDOn0g,529
 versionhq/agent/TEMPLATES/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/agent_network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/agent_network/formation.py,sha256=jXCc9dS9gOgjNoPNoqKaqKPrHqmcFIvR2JwKYTgiQW0,7505
-versionhq/agent_network/model.py,sha256=CYkIU1W1Ijh5DQD18vRcD1g8myZWmqAYNS0PlRnEX-o,15899
+versionhq/agent_network/formation.py,sha256=P5YEVAVwxiAUQ5J4mW3sK-WTd0nuXLo_LOFiKz1Kqik,7313
+versionhq/agent_network/model.py,sha256=eDZCr628mVYRGKtcyDVWHlszW8wmq7A6DpKIhxCn6Vo,15860
 versionhq/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/clients/customer/__init__.py,sha256=-YXh1FQfvpfLacK8SUC7bD7Wx_eIEi4yrkCC_cUasFg,217
+versionhq/clients/customer/__init__.py,sha256=2hDrX3xyEMgrH1E3SjZVkd4V8cR3HWUjhDt7lpThJsM,218
 versionhq/clients/customer/model.py,sha256=_AtaVVMm9MgCwrQ-HTRQ2oXUMKrSCEfZwE2JdRz3xTw,2508
 versionhq/clients/product/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/clients/product/model.py,sha256=3w__pug9XRe4LIm9wX8C8WKqi40r081Eb1q2vWk9UaU,3694
+versionhq/clients/product/model.py,sha256=Beb95U4XfdvCbSi5mT6j3-P8D_dforF-H04M9bnUWlE,3588
 versionhq/clients/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/clients/workflow/model.py,sha256=_yCbmwzexjQqEGli7XX0vFG6yFnAOe9QDoMWQaU_qZE,3920
+versionhq/clients/workflow/model.py,sha256=AI8pAO2MW-ysxC2TDgPPQ-5O4HrhSBG9QjfuaJb9Rw4,3917
 versionhq/knowledge/__init__.py,sha256=qW7IgssTA4_bFFV9ziOcYRfGjlq1c8bkb-HnfWknpuQ,567
 versionhq/knowledge/_utils.py,sha256=YWRF8U533cfZes_gZqUvdj-K24MD2ri1R0gjc_aPYyc,402
 versionhq/knowledge/embedding.py,sha256=KfHc__1THxb5jrg1EMrF-v944RDuIr2hE0l-MtM3Bp0,6826
@@ -36,7 +38,7 @@ versionhq/knowledge/source.py,sha256=-hEUPtJUHHMx4rUKtiHl19J8xAMw-WVBw34zwa2jZ08
 versionhq/knowledge/source_docling.py,sha256=XpavmLvh4dLcuTikj8MCE9KG52oQMafy7_wBneliMK0,4994
 versionhq/knowledge/storage.py,sha256=Kd-4r6aWM5EDaoXrzKXbgi1hY6tysSQARPGXM95qMmU,8266
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/llm/llm_vars.py,sha256=fl9MOjPzBEIymp99BNLD39VmB7rQ9L4YDnfLNXbUZws,5896
+versionhq/llm/llm_vars.py,sha256=gc3igJdjhY4osnN75STEeELuWiPwlA1S-4JCqK5RQkE,5894
 versionhq/llm/model.py,sha256=m4OaFgGWKCZjmfN-OY3KoqG1K4T31UF7QVkYUcnpjdg,17273
 versionhq/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/memory/contextual_memory.py,sha256=QEMVvHuEXxY7M6-12S8HhyFKf108KfX8Zzt7paPW048,3882
@@ -46,28 +48,34 @@ versionhq/storage/base.py,sha256=p-Jas0fXQan_qotnRD6seQxrT2lj-uw9-SmHQhdppcs,355
 versionhq/storage/ltm_sqlite_storage.py,sha256=LeJE4ZPUWjyY1E5nNCHoKujTHFDR2BO_LAMvAOX-WHg,3957
 versionhq/storage/mem0_storage.py,sha256=ZY8MELBWaINRv9YuRW5MxH7dj2cII-L0i3xSD6o1-2M,3781
 versionhq/storage/rag_storage.py,sha256=bS2eE874obarYl-4hT6ZWYWTRsqtfuGpKgKzERmM6Uo,7433
-versionhq/storage/task_output_storage.py,sha256=nkBNmBbrQeEgds3lAyKl4ugDWLtWRoCQUO6KiOmCIMU,5362
+versionhq/storage/task_output_storage.py,sha256=9x76i_9nXuapujX4Fj1Jz1h-bF9w7DzUbzdJIPx9pPo,5374
 versionhq/storage/utils.py,sha256=r5ghA_ktdR2IuzlzKqZYCjsNxztEMzyhWLneA4cFuWY,748
 versionhq/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/task/evaluation.py,sha256=qQSA5ZWTWA3he54ystsYpTKXJWv68gBL6DCq8ZW1bl8,3813
 versionhq/task/formatter.py,sha256=N8Kmk9vtrMtBdgJ8J7RmlKNMdZWSmV8O1bDexmCWgU0,643
-versionhq/task/model.py,sha256=C5QttgJrfhx0UL82845ZBIhhc8oLiQkAWz2i6UkskMM,29023
+versionhq/task/model.py,sha256=lTpog4ERVlcl95ku-ZCVbi7htT7hnI-YGM_JFZJ27k8,29462
 versionhq/task/structured_response.py,sha256=tqOHpch8CVmMj0aZXjdDWtPNcVmBW8DVZnBvPBwS4PM,5053
 versionhq/task/TEMPLATES/Description.py,sha256=hKhpbz0ztbkUMXz9KiL-P40fis9OB5ICOdL9jCtgAhU,864
 versionhq/task_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/task_graph/colors.py,sha256=naJCx4Vho4iuJtbW8USUXb-M5uYvd5ds2p8qbjUfRus,669
 versionhq/task_graph/draft.py,sha256=mVhCHDH-7N-SQRssE50KGIAgd9gdvdeWjt8ofm-SYI4,4943
-versionhq/task_graph/model.py,sha256=l4Alvdtdl-fwYG7eMo655HF0zx1HkKRiPiST_Ra7hzg,29305
+versionhq/task_graph/model.py,sha256=2FsZj6v7BGwUgf1RAp2oohz9tf91swIij0j6l0CEe6E,29364
 versionhq/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/tool/cache_handler.py,sha256=iL8FH7X0G-cdT0uhJwzuhLDaadTXOdfybZcDy151-es,1085
-versionhq/tool/composio_tool.py,sha256=IATfsEnF_1RPJyGtPBmAtEJh5XPcgDHpyG3SUR461Og,8572
-versionhq/tool/composio_tool_vars.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtgpqOzKZQ,1843
 versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1205
 versionhq/tool/model.py,sha256=ve9C4WyiRjQigOU0hRWVxtSUWAQNntlmeW-_DL0_lJY,12328
 versionhq/tool/rag_tool.py,sha256=dW5o-83V4bMFFJEj3PUm7XjblwrYJGmZVBlCpPj6CeM,3852
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.2.4.5.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.4.5.dist-info/METADATA,sha256=gwFpoEsZC3Jfo5k2bvpZBPix_pWZCJ0h_UDDgRfrsBw,21146
-versionhq-1.2.4.5.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-versionhq-1.2.4.5.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.4.5.dist-info/RECORD,,
+versionhq/tool/composio/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+versionhq/tool/composio/model.py,sha256=GIFKso_e_4a3BdaulqU_i6Y9JFAExNBjzHUHR_zZeSI,8577
+versionhq/tool/composio/params.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtgpqOzKZQ,1843
+versionhq/tool/gpt/__init__.py,sha256=A6xCuf_GUBs7wfx904J_Vd2t1GJCcf0lMKOL7MbZce4,160
+versionhq/tool/gpt/_enum.py,sha256=fw31aYeQBY8vfRwzjvwa1IVhhpAAQt9m1jqQTjTBpLk,500
+versionhq/tool/gpt/cup.py,sha256=Vz7VmTQV0IvUZcyN9znH7d7XUvLzm2SlSXJ5NoJFk3Q,5048
+versionhq/tool/gpt/file_search.py,sha256=CoPwNNoGF4U-0OuUMixqjdt5YoO0tqWbdbR_7UE-dMk,5597
+versionhq/tool/gpt/web_search.py,sha256=NoeW7m_YxhUp8NanTjBt3GX6nAU6G8bi_jfpLfWyx90,3127
+versionhq-1.2.4.6.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.4.6.dist-info/METADATA,sha256=clqmU4LOEF01NFhvqX6NQTB8poWzPYDKrTzbgImprV0,21146
+versionhq-1.2.4.6.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+versionhq-1.2.4.6.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.4.6.dist-info/RECORD,,