naas-abi 1.0.3.dev1__py3-none-any.whl → 1.0.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. naas_abi/models/default.py +1 -0
  2. naas_abi/models/gpt_4_1.py +26 -0
  3. naas_abi/pipelines/RemoveIndividualPipeline.py +2 -1
  4. {naas_abi-1.0.3.dev1.dist-info → naas_abi-1.0.4.dist-info}/METADATA +1 -1
  5. {naas_abi-1.0.3.dev1.dist-info → naas_abi-1.0.4.dist-info}/RECORD +7 -6
  6. {naas_abi-1.0.3.dev1.dist-info → naas_abi-1.0.4.dist-info}/WHEEL +0 -0
  7. {naas_abi-1.0.3.dev1.dist-info → naas_abi-1.0.4.dist-info}/top_level.txt +0 -0
naas_abi/models/default.py CHANGED
@@ -22,3 +22,4 @@ def get_model() -> ChatModel:
22
22
  )
23
23
 
24
24
  return cloud_model
25
+ return cloud_model
naas_abi/models/gpt_4_1.py ADDED
@@ -0,0 +1,26 @@
1
+ from langchain_openai import ChatOpenAI
2
+ from naas_abi import ABIModule
3
+ from naas_abi_core.models.Model import ChatModel
4
+ from pydantic import SecretStr
5
+
6
+ ID = "gpt-4.1"
7
+ NAME = "gpt-4.1"
8
+ DESCRIPTION = "GPT-4.1 excels at instruction following and tool calling, with broad knowledge across domains. It features a 1M token context window, and low latency without a reasoning step."
9
+ IMAGE = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi-demo/ontology_ABI.png"
10
+ CONTEXT_WINDOW = 1047576
11
+ OWNER = "openai"
12
+
13
+ model: ChatModel = ChatModel(
14
+ model_id=ID,
15
+ provider=OWNER,
16
+ name=NAME,
17
+ description=DESCRIPTION,
18
+ image=IMAGE,
19
+ owner=OWNER,
20
+ model=ChatOpenAI(
21
+ model=ID,
22
+ temperature=0,
23
+ api_key=SecretStr(ABIModule.get_instance().configuration.openai_api_key),
24
+ ),
25
+ context_window=CONTEXT_WINDOW,
26
+ )
naas_abi/pipelines/RemoveIndividualPipeline.py CHANGED
@@ -3,7 +3,8 @@ from enum import Enum
3
3
  from typing import Annotated, List
4
4
 
5
5
  from langchain_core.tools import BaseTool, StructuredTool
6
- from naas_abi import ABIModule, logger
6
+ from naas_abi import ABIModule
7
+ from naas_abi_core import logger
7
8
  from naas_abi_core.pipeline import Pipeline, PipelineConfiguration, PipelineParameters
8
9
  from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStoreService
9
10
  from naas_abi_core.utils.Expose import APIRouter
{naas_abi-1.0.3.dev1.dist-info → naas_abi-1.0.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: naas-abi
3
- Version: 1.0.3.dev1
3
+ Version: 1.0.4
4
4
  Summary: Add your description here
5
5
  Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
6
6
  Requires-Python: <4,>=3.10
{naas_abi-1.0.3.dev1.dist-info → naas_abi-1.0.4.dist-info}/RECORD RENAMED
@@ -18,7 +18,8 @@ naas_abi/apps/terminal_agent/main.py,sha256=7MN88mivIV7gP1lELnqNxP_huwTbavhnTjkN
18
18
  naas_abi/apps/terminal_agent/terminal_style.py,sha256=YOpfvBlKU52wNyGEKbZPiRQVKxug-hI92H8ScV5L8Ew,5254
19
19
  naas_abi/models/airgap_gemma.py,sha256=I9pWN_DSgy4h6rT7OiYWDVijyTMoEgONm4xfZYWvp6I,8469
20
20
  naas_abi/models/airgap_qwen.py,sha256=EQZ-0tAKZCTCqnovLiGCx4ss0yh2Jjk0xh7wCwDomG8,682
21
- naas_abi/models/default.py,sha256=q8vVS0CL46vQOq_SE_KeCa0pLY-zglDFe0ssUeMTq9I,710
21
+ naas_abi/models/default.py,sha256=R7QmR3-IrNwvsm_l0NDhawrWqTYDY6S_w0cr6TbHic4,737
22
+ naas_abi/models/gpt_4_1.py,sha256=0luDssCvt8BoK_iVDRI7wkZ5GYn1d--G3GtnN6jiPIg,838
22
23
  naas_abi/pipelines/AIAgentOntologyGenerationPipeline.py,sha256=QWrooZ_xJy0uX92Tzw7fBOs_uME2nLhsjeWi0EGTUBA,24640
23
24
  naas_abi/pipelines/AIAgentOntologyGenerationPipeline_test.py,sha256=lDsuZKfQlCtLNKQ09nkSHemsvjSNdeZAhaDw-cPYbjo,4130
24
25
  naas_abi/pipelines/AddIndividualPipeline.py,sha256=EKrJX4ODXBMSFUl0R2UsaSmLaj0cUJ3AdIhhtU0Fhso,8445
@@ -27,7 +28,7 @@ naas_abi/pipelines/InsertDataSPARQLPipeline.py,sha256=yo-GaAW4Od1503iG_3BHCUzRE3
27
28
  naas_abi/pipelines/InsertDataSPARQLPipeline_test.py,sha256=AdYonpWm2oL4CfJkAtFQRb3XPyFiGCqaxwsfx0vr1yg,3250
28
29
  naas_abi/pipelines/MergeIndividualsPipeline.py,sha256=PMCU_f0u689I6l93-VpgNkGgWxeUByLVpakVKmFG1rY,9206
29
30
  naas_abi/pipelines/MergeIndividualsPipeline_test.py,sha256=G__YPaHPAup3_PwAsjy3qJWmQNYFXUjl4BvgqsYw2uw,2995
30
- naas_abi/pipelines/RemoveIndividualPipeline.py,sha256=9-zAjV9tgOx-xMeBsJceNZNWEuaL9P4XLWbaTJ9PMdI,5533
31
+ naas_abi/pipelines/RemoveIndividualPipeline.py,sha256=rgoel5Gbtl99E2jlDzBr7-bMk7Wgd0mLuuf-GOzz_Ys,5558
31
32
  naas_abi/pipelines/RemoveIndividualPipeline_test.py,sha256=RMxBoDFcU7eHJsmK7kjnhXNLlFrJE7KgwHoJez-Fvto,1702
32
33
  naas_abi/pipelines/UpdateCommercialOrganizationPipeline.py,sha256=TDboVTlOpsbrixRkp-b9dA6Cm194D4ZVUOeQX_uSMe4,6768
33
34
  naas_abi/pipelines/UpdateDataPropertyPipeline.py,sha256=m6wef4tVfBllt4pkflkkKB_Ncw9aEkTHV75ADvUjZcE,6723
@@ -55,7 +56,7 @@ naas_abi/workflows/GetSubjectGraphWorkflow.py,sha256=EkgrFQPyaUVeC8sd7oVcxT9LUka
55
56
  naas_abi/workflows/GetSubjectGraphWorkflow_test.py,sha256=fzzFYUwojM39x5Ld-59jHMDbGy4Lcq6uO8_BAvZV0ug,1886
56
57
  naas_abi/workflows/SearchIndividualWorkflow.py,sha256=6vWwywKNBlaosopPxiw0OpQnrn99wuiji0uA7jwKOn4,6572
57
58
  naas_abi/workflows/SearchIndividualWorkflow_test.py,sha256=KBhd58TATMSWWdThYPg7veAySR4hts5LuoygQqQIiaU,2749
58
- naas_abi-1.0.3.dev1.dist-info/METADATA,sha256=QrIQ2gSbXQ1RjZSk-k2n4yRNiXtv-yZ6y5uXhezaEzc,363
59
- naas_abi-1.0.3.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
60
- naas_abi-1.0.3.dev1.dist-info/top_level.txt,sha256=Dhf3mhATB9-JuvBR8Y55OabvI-PCeQUqb5ae3EMmbeA,9
61
- naas_abi-1.0.3.dev1.dist-info/RECORD,,
59
+ naas_abi-1.0.4.dist-info/METADATA,sha256=W0CZpRgs9iCHaM1Zw7mNHAJLOIqs-i7DFiZB06MebiM,358
60
+ naas_abi-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
61
+ naas_abi-1.0.4.dist-info/top_level.txt,sha256=Dhf3mhATB9-JuvBR8Y55OabvI-PCeQUqb5ae3EMmbeA,9
62
+ naas_abi-1.0.4.dist-info/RECORD,,