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

naas_abi/models/default.py

@@ -22,4 +22,3 @@ def get_model() -> ChatModel:
         )
 
         return cloud_model
-        return cloud_model

{naas_abi-1.0.3.dist-info → naas_abi-1.0.3.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi
-Version: 1.0.3
+Version: 1.0.3.dev1
 Summary: Add your description here
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 Requires-Python: <4,>=3.10

{naas_abi-1.0.3.dist-info → naas_abi-1.0.3.dev1.dist-info}/RECORD

@@ -18,8 +18,7 @@ naas_abi/apps/terminal_agent/main.py,sha256=7MN88mivIV7gP1lELnqNxP_huwTbavhnTjkN
 naas_abi/apps/terminal_agent/terminal_style.py,sha256=YOpfvBlKU52wNyGEKbZPiRQVKxug-hI92H8ScV5L8Ew,5254
 naas_abi/models/airgap_gemma.py,sha256=I9pWN_DSgy4h6rT7OiYWDVijyTMoEgONm4xfZYWvp6I,8469
 naas_abi/models/airgap_qwen.py,sha256=EQZ-0tAKZCTCqnovLiGCx4ss0yh2Jjk0xh7wCwDomG8,682
-naas_abi/models/default.py,sha256=R7QmR3-IrNwvsm_l0NDhawrWqTYDY6S_w0cr6TbHic4,737
-naas_abi/models/gpt_4_1.py,sha256=0luDssCvt8BoK_iVDRI7wkZ5GYn1d--G3GtnN6jiPIg,838
+naas_abi/models/default.py,sha256=q8vVS0CL46vQOq_SE_KeCa0pLY-zglDFe0ssUeMTq9I,710
 naas_abi/pipelines/AIAgentOntologyGenerationPipeline.py,sha256=QWrooZ_xJy0uX92Tzw7fBOs_uME2nLhsjeWi0EGTUBA,24640
 naas_abi/pipelines/AIAgentOntologyGenerationPipeline_test.py,sha256=lDsuZKfQlCtLNKQ09nkSHemsvjSNdeZAhaDw-cPYbjo,4130
 naas_abi/pipelines/AddIndividualPipeline.py,sha256=EKrJX4ODXBMSFUl0R2UsaSmLaj0cUJ3AdIhhtU0Fhso,8445
@@ -56,7 +55,7 @@ naas_abi/workflows/GetSubjectGraphWorkflow.py,sha256=EkgrFQPyaUVeC8sd7oVcxT9LUka
 naas_abi/workflows/GetSubjectGraphWorkflow_test.py,sha256=fzzFYUwojM39x5Ld-59jHMDbGy4Lcq6uO8_BAvZV0ug,1886
 naas_abi/workflows/SearchIndividualWorkflow.py,sha256=6vWwywKNBlaosopPxiw0OpQnrn99wuiji0uA7jwKOn4,6572
 naas_abi/workflows/SearchIndividualWorkflow_test.py,sha256=KBhd58TATMSWWdThYPg7veAySR4hts5LuoygQqQIiaU,2749
-naas_abi-1.0.3.dist-info/METADATA,sha256=RZGiFr_xokgIbf8X8V6kiJoyHmFyZEVVST49CueAooM,358
-naas_abi-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-naas_abi-1.0.3.dist-info/top_level.txt,sha256=Dhf3mhATB9-JuvBR8Y55OabvI-PCeQUqb5ae3EMmbeA,9
-naas_abi-1.0.3.dist-info/RECORD,,
+naas_abi-1.0.3.dev1.dist-info/METADATA,sha256=QrIQ2gSbXQ1RjZSk-k2n4yRNiXtv-yZ6y5uXhezaEzc,363
+naas_abi-1.0.3.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+naas_abi-1.0.3.dev1.dist-info/top_level.txt,sha256=Dhf3mhATB9-JuvBR8Y55OabvI-PCeQUqb5ae3EMmbeA,9
+naas_abi-1.0.3.dev1.dist-info/RECORD,,

naas_abi/models/gpt_4_1.py

@@ -1,26 +0,0 @@
-from langchain_openai import ChatOpenAI
-from naas_abi import ABIModule
-from naas_abi_core.models.Model import ChatModel
-from pydantic import SecretStr
-
-ID = "gpt-4.1"
-NAME = "gpt-4.1"
-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."
-IMAGE = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi-demo/ontology_ABI.png"
-CONTEXT_WINDOW = 1047576
-OWNER = "openai"
-
-model: ChatModel = ChatModel(
-    model_id=ID,
-    provider=OWNER,
-    name=NAME,
-    description=DESCRIPTION,
-    image=IMAGE,
-    owner=OWNER,
-    model=ChatOpenAI(
-        model=ID,
-        temperature=0,
-        api_key=SecretStr(ABIModule.get_instance().configuration.openai_api_key),
-    ),
-    context_window=CONTEXT_WINDOW,
-)