traia-iatp 0.1.11__py3-none-any.whl → 0.1.13__py3-none-any.whl

traia_iatp/core/models.py

@@ -21,7 +21,7 @@ class MCPServer(BaseModel):
     capabilities: List[str] = Field(default_factory=list, description="List of capabilities/APIs")
     metadata: Dict[str, Any] = Field(default_factory=dict, description="Additional metadata")
     
-    class Config:
+    class ConfigDict:
         json_encoders = {HttpUrl: str}
 
 
@@ -113,7 +113,7 @@ class UtilityAgent(BaseModel):
     updated_at: datetime = Field(default_factory=datetime.utcnow)
     metadata: Dict[str, Any] = Field(default_factory=dict, description="Additional metadata")
     
-    class Config:
+    class ConfigDict:
         json_encoders = {HttpUrl: str, datetime: lambda v: v.isoformat()}
 
 
@@ -143,7 +143,7 @@ class UtilityAgentRegistryEntry(BaseModel):
     last_health_check: Optional[datetime] = Field(None, description="Last successful health check")
     is_active: bool = Field(default=True, description="Whether the agent is active")
     
-    class Config:
+    class ConfigDict:
         json_encoders = {HttpUrl: str, datetime: lambda v: v.isoformat()}
 
 

traia_iatp/mcp/mcp_agent_template.py

@@ -335,7 +335,8 @@ def run_with_mcp_tools(
                 agents=agents,
                 tasks=tasks,
                 verbose=verbose,
-                process=process
+                process=process,
+                tracing=True if os.getenv("AGENTOPS_API_KEY") else False,
             )
             
             # Kickoff the crew with inputs

traia_iatp/mcp/templates/pyproject.toml.j2

@@ -4,10 +4,10 @@ version = "0.1.0"
 description = "MCP server for {{ api_name }} API{% if auth_description %} with authentication support{% endif %}"
 requires-python = ">=3.12"
 dependencies = [
-    "fastmcp>=2.10.1",
+    "fastmcp>=2.12.5",
     "python-dotenv>=1.1.1",
-    "requests>=2.32.4",
-    "starlette>=0.46.2",
+    "requests>=2.32.5",
+    "starlette>=0.48.0",
     "retry>=0.9.2",
 {% if sdk_package %}
     "{{ sdk_package }}",

traia_iatp/server/templates/agent.py.j2

@@ -25,8 +25,15 @@ except ImportError:
     def operation(func):
         return func
 
+DEFAULT_LLM = LLM(
+    model=os.getenv("LLM_MODEL", "gpt-4.1-nano"),  # Using environment variable with fallback
+    temperature=os.getenv("LLM_MODEL_TEMPERATURE", 0.1),
+    api_key=os.getenv("OPENAI_API_KEY")
+)
+
 logger = logging.getLogger(__name__)
 
+logger.info(f"Current LLM model used: {os.getenv("LLM_MODEL", "gpt-4.1-nano")}")
 
 class {{ class_name }}Agent:
     """{{ agent_name }} agent that processes requests using {{ mcp_server_name }}."""
@@ -61,6 +68,7 @@ class {{ class_name }}Agent:
             ),
             verbose=True,
             allow_delegation=False,
+            llm=DEFAULT_LLM,
             tools_subset=tools_subset
         )
     

traia_iatp/server/templates/agent_executor.py.j2

@@ -17,7 +17,7 @@ from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events.event_queue import EventQueue
 from a2a.types import Message, TextPart
 from a2a.utils import new_agent_text_message
-from crewai import Task
+from crewai import Task, LLM
 from traia_iatp.mcp import MCPServerConfig, MCPAgentBuilder, run_with_mcp_tools, MCPServerInfo
 
 # Import AgentOps for operation tracking
@@ -32,8 +32,17 @@ except ImportError:
     def operation(func):
         return func
 
+
+DEFAULT_LLM = LLM(
+    model=os.getenv("LLM_MODEL", "gpt-4.1-nano"),  # Using environment variable with fallback
+    temperature=os.getenv("LLM_MODEL_TEMPERATURE", 0.1),
+    api_key=os.getenv("OPENAI_API_KEY")
+)
+
 logger = logging.getLogger(__name__)
 
+logger.info(f"Current LLM model used: {os.getenv("LLM_MODEL", "gpt-4.1-nano")}")
+
 # Create a thread pool for CPU-bound CrewAI operations
 executor = ThreadPoolExecutor(max_workers=10)
 
@@ -129,7 +138,8 @@ class {{ class_name }}AgentExecutor(AgentExecutor):
         agent = MCPAgentBuilder.create_agent(
             role=f"{{ agent_name }} Specialist",
             goal=f"Process the request using {self.mcp_config.name} capabilities",
-            backstory=f"You are an expert at using {self.mcp_config.name}. {self.mcp_config.description}"
+            backstory=f"You are an expert at using {self.mcp_config.name}. {self.mcp_config.description}",
+            llm=DEFAULT_LLM
         )
         
         # Create a task

traia_iatp/server/templates/env.example.j2

@@ -9,7 +9,10 @@ HOST=0.0.0.0
 AGENTOPS_API_KEY=your-agentops-api-key-here
 
 # Language Model Configuration
-LLM_MODEL=openai/gpt-4.1
+LLM_MODEL=your-llm-model-name
+LLM_MODEL_TEMPERATURE=your-llm-model-temperature
+EMBEDDINGS_OPENAI_MODEL_NAME=your-embedding-model
+
 # You can also use other models:
 # LLM_MODEL=vertex_ai/gemini-2.0-flash
 # LLM_MODEL=anthropic/claude-4-sonnetf

traia_iatp/server/templates/pyproject.toml.j2

@@ -5,18 +5,18 @@ description = "{{ agent_description }}"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
-    "crewai>=0.130.0",
-    "crewai-tools[mcp]>=0.47.1",
-    "agentops>=0.3.0",
+    "crewai>=0.203.1",
+    "crewai-tools[mcp]>=0.76.0",
+    "agentops>=0.4.21",
     "a2a-sdk>=0.2.6",
-    "fastmcp>=2.8.1",
+    "fastmcp>=2.12.5",
     "hypercorn>=0.17.3",  # HTTP/2 support via h2 dependency
     "httpx[http2]>=0.28.1",  # HTTP/2 support with http2 extra
-    "starlette>=0.46.2",  # For SSE streaming support
-    "sse-starlette>=2.3.6",  # Server-Sent Events support
-    "pydantic>=2.11.0",
-    "python-dotenv>=1.1.0",
-    "pymongo>=4.13.0",  # Required for MCP registry
+    "starlette>=0.48.0",  # For SSE streaming support
+    "sse-starlette>=3.0.2",  # Server-Sent Events support
+    "pydantic>=2.12.2",
+    "python-dotenv>=1.1.1",
+    "pymongo>=4.15.3",  # Required for MCP registry
     "traia-iatp",  # Core IATP functionality - always use latest version
     {% for dep in additional_dependencies %}
     "{{ dep }}",
@@ -30,8 +30,7 @@ grpc = [
     "grpcio-reflection>=1.71.0",
 ]
 
-[tool.uv]
-dev-dependencies = [
+dev = [
     "pytest>=7.0.0",
     "pytest-asyncio>=0.21.0",
     "black>=23.0.0",

{traia_iatp-0.1.11.dist-info → traia_iatp-0.1.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.11
+Version: 0.1.13
 Summary: Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools
 Project-URL: Documentation, https://pypi.org/project/traia-iatp
 Project-URL: Source, https://github.com/Traia-IO/IATP
@@ -12,31 +12,35 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: a2a-sdk==0.2.6
 Requires-Dist: aiohttp>=3.12.13
-Requires-Dist: crewai>=0.130.0
-Requires-Dist: crewai-tools[mcp]>=0.47.1
+Requires-Dist: crewai>=0.203.1
+Requires-Dist: crewai-tools[mcp]>=0.76.0
 Requires-Dist: docker>=7.1.0
-Requires-Dist: fastapi>=0.115.12
-Requires-Dist: fastmcp>=2.8.1
-Requires-Dist: httpx>=0.28.1
+Requires-Dist: fastapi>=0.119.0
+Requires-Dist: fastmcp>=2.12.5
+Requires-Dist: httpx[http2]>=0.28.1
 Requires-Dist: jinja2>=3.1.6
-Requires-Dist: openai>=1.90.0
-Requires-Dist: pydantic>=2.11.5
-Requires-Dist: pymongo[aws]>=4.13.2
-Requires-Dist: python-dotenv>=1.1.0
+Requires-Dist: openai>=1.109.1
+Requires-Dist: pydantic>=2.12.2
+Requires-Dist: pymongo[aws]>=4.15.3
+Requires-Dist: python-dotenv>=1.1.1
 Requires-Dist: pytz>=2025.2
-Requires-Dist: requests>=2.32.3
-Requires-Dist: rich>=13.9.4
-Requires-Dist: typer>=0.16.0
-Requires-Dist: uvicorn>=0.34.3
-Requires-Dist: agentops>=0.3.0
-Requires-Dist: dnspython>=2.7.0
+Requires-Dist: requests>=2.32.5
+Requires-Dist: rich>=14.2.0
+Requires-Dist: typer>=0.19.2
+Requires-Dist: uvicorn>=0.37.0
+Requires-Dist: agentops>=0.4.21
+Requires-Dist: dnspython>=2.8.0
 Provides-Extra: dev
-Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest>=8.4.2; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
 Requires-Dist: black>=23.0.0; extra == "dev"
 Requires-Dist: flake8>=6.0.0; extra == "dev"
 Requires-Dist: mypy>=1.0.0; extra == "dev"
 Requires-Dist: pre-commit>=3.0.0; extra == "dev"
+Provides-Extra: publish
+Requires-Dist: wheel; extra == "publish"
+Requires-Dist: twine; extra == "publish"
+Requires-Dist: build; extra == "publish"
 Dynamic: license-file
 
 # Traia IATP

{traia_iatp-0.1.11.dist-info → traia_iatp-0.1.13.dist-info}/RECORD

@@ -9,10 +9,10 @@ traia_iatp/client/crewai_a2a_tools.py,sha256=m989j13d8Ba1SDDNXuprBA0W7pjXvVApcCJ
 traia_iatp/client/grpc_a2a_tools.py,sha256=pVt6UASXdt096ZNfcTqY0pjaTqLr-0Ptl8xd68rxRWQ,13003
 traia_iatp/client/root_path_a2a_client.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 traia_iatp/core/__init__.py,sha256=SdGs25MkNWok77wRtbCJFGkD85QAmYUAB40Fn9mHeXk,755
-traia_iatp/core/models.py,sha256=nvpLJjXAW2zTLOejjFXaTeDEN_gRo6lETGOCyvYKXPI,8381
+traia_iatp/core/models.py,sha256=YLyVwwtfT8OwrFqmgsEMGYVbyoBgHm-VPYvt3VOcLhI,8393
 traia_iatp/mcp/__init__.py,sha256=kb2bX9yMwatWBFb0xq-BeuDD_d7DLRt8huA32hDcnCo,427
 traia_iatp/mcp/client.py,sha256=IMw_fO36GcAa92zHOkCuJJquG9untnkgX6LPOtrN1lU,7859
-traia_iatp/mcp/mcp_agent_template.py,sha256=vGiBJy7ZtQ4Z0KkoHMW2lrmow2MAoIMpY9cjmjYylrU,14997
+traia_iatp/mcp/mcp_agent_template.py,sha256=0liaDYm2NXEubneE6bZfEgwJfUTpL_T0D-lsxpJgNYY,15072
 traia_iatp/mcp/traia_mcp_adapter.py,sha256=0WIVERdWk2Bnx_mVwito66blT6gstI4RKiP_lP4VoZI,11945
 traia_iatp/mcp/templates/Dockerfile.j2,sha256=bYCbpsawIO8WfGPjncvE8pZ6Dn9US6dsK61WdpW5DN4,1332
 traia_iatp/mcp/templates/README.md.j2,sha256=xxdrFYcY8RmfKzSJs87PGSxRFjkGcrHzfY9a66wN_LI,5781
@@ -22,7 +22,7 @@ traia_iatp/mcp/templates/docker-compose.yml.j2,sha256=Zge8VYa629Av8E22asCV60jlrF
 traia_iatp/mcp/templates/dockerignore.j2,sha256=EsXzYQ5Ku3NROEgmfv670VuSpla8_-tllblEE0Hp89Y,366
 traia_iatp/mcp/templates/gitignore.j2,sha256=shz2n7MlRBOvwRm4leCjpxDi8r0bm8DJLAtvLcp6X_0,698
 traia_iatp/mcp/templates/mcp_health_check.py.j2,sha256=s_7C1fUdApoto52HxX1BN2PNLaEnr00B9ZNw_bpcPDg,5234
-traia_iatp/mcp/templates/pyproject.toml.j2,sha256=raBTKjK0tSoWuAO39MBLHqsvZzu4kXjHw012qJKhQyM,568
+traia_iatp/mcp/templates/pyproject.toml.j2,sha256=ZGUFtKN_c3OSaaw6lXekEBegkufkKxdW7tvGKCl6CGY,568
 traia_iatp/mcp/templates/run_local_docker.sh.j2,sha256=n0dwabQMgQe1pDpqCZmZafVtebv97M2E21u5N8SQxoc,3233
 traia_iatp/mcp/templates/server.py.j2,sha256=zty-7UGU8CgWSbbFj-3tuJfrGJI79pBBYhcXDzU5GZA,8169
 traia_iatp/registry/__init__.py,sha256=YGFGRqgRp0f09ND-FWTG9GQNwxz-HpUBYbt_KhLP-ZY,662
@@ -49,14 +49,14 @@ traia_iatp/server/templates/README.md,sha256=wz1d0dhKvvHXyfTxM3QBDJJE-RdLN1yxCUQ
 traia_iatp/server/templates/README.md.j2,sha256=yy7N99FQ2ZkS2I-O2_YO9_UmOP3mLGNI3vpKrIqjnig,10625
 traia_iatp/server/templates/__init__.py,sha256=OfnDqxbgP51KEUSxFpfnePOF_72LlC00DOeR744ky4M,53
 traia_iatp/server/templates/__main__.py.j2,sha256=fvVIrQR4zUGmdRv6i1p2QQET6nPntDAS7xZiJD1ZOGg,20756
-traia_iatp/server/templates/agent.py.j2,sha256=cp1E9NBXZnWCs9UkP_4sB6cAEu6_srv2OQNN_7evweI,3389
+traia_iatp/server/templates/agent.py.j2,sha256=rRaVXsVZYuTdKkZ9mldiLWxu5KIfSpRpKIgX3oX7YVU,3712
 traia_iatp/server/templates/agent_config.json.j2,sha256=YGikrYXcVKZfZJP9OnlS-GJXaGExlaRN2DB_Lo5Pr7k,797
-traia_iatp/server/templates/agent_executor.py.j2,sha256=bV3DG0_3fLZP_n2-L8u0sRETi02NyvMF0dBiIdnFH6U,11203
+traia_iatp/server/templates/agent_executor.py.j2,sha256=cTSU-rTRtUMPapy4NalQPwfGGaWIv4tLRPnS1C7cZSc,11533
 traia_iatp/server/templates/docker-compose.yml.j2,sha256=3jRkkUU6gazKyVgelfQRf6x0KRbpXINEsVkBIgqrQbc,610
-traia_iatp/server/templates/env.example.j2,sha256=PEk8G0Itq8ombTCqqJzVnVxgAvkJjkILzb_r_3Pzc64,1888
+traia_iatp/server/templates/env.example.j2,sha256=7IVLLmk_VWN6jHd0MCeQlb_rg7eaZhpM6ruaavXTRFc,1993
 traia_iatp/server/templates/gitignore.j2,sha256=rXu4kNnZUFgBDbLsburNohFVAjyE9xEeHbpEcqGGSB8,740
 traia_iatp/server/templates/grpc_server.py.j2,sha256=GFZv0b146QthPDnBMEn2GpD8fRCfuCXzcq4RBEEGAac,7610
-traia_iatp/server/templates/pyproject.toml.j2,sha256=kHA3YilqoeUIF81UTBWpBVduFBXbVHdC_U5iuunTpKY,1832
+traia_iatp/server/templates/pyproject.toml.j2,sha256=pO3MYT6OXHrYXv6ARxHR7_bNs2PO3qEgsqxDddcWjWA,1811
 traia_iatp/server/templates/run_local_docker.sh.j2,sha256=4h1GGtwM8gsSDbChblsea2muS0MYXZtJwY4swso1PyI,3495
 traia_iatp/server/templates/server.py.j2,sha256=uHQfngwj2IDbEjczvsX2SnikYNAxT-yJEKbQl456gFs,6351
 traia_iatp/special_agencies/__init__.py,sha256=9Uh0X8gjrAnl6yjmjGeCaxyJvB3HVwoOKhTc9ljmsr4,122
@@ -65,9 +65,9 @@ traia_iatp/utils/__init__.py,sha256=D739Fc3KWPlAaPZ-3XM7uRvBAA7V8b4t6PBYJvMSsig,
 traia_iatp/utils/docker_utils.py,sha256=mPD09TPJKEe9SKjrR5E_o9z8xle-iCZq6JLzHZ6JXFk,8634
 traia_iatp/utils/general.py,sha256=g6sRKzalBPwRWobBg3a7pz4eNIvvxv8RefoUNFdHm4o,1503
 traia_iatp/utils/iatp_utils.py,sha256=RrGzk2ENvGIwuuMigOUrK5Yne0esl1r_K2P2iVygUgM,4680
-traia_iatp-0.1.11.dist-info/licenses/LICENSE,sha256=R8vcu8GZdTTXF40sbOll11-FbarDm7_PiuaGzQp9whw,1065
-traia_iatp-0.1.11.dist-info/METADATA,sha256=_WtfDrIpDsF62i8u4bQ8PxSO0INIgARJqnn50eqPhrI,13256
-traia_iatp-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-traia_iatp-0.1.11.dist-info/entry_points.txt,sha256=K4p9Z2lBdPEybzXfed3JbUtrVHDdg60CRnJ178df7K0,55
-traia_iatp-0.1.11.dist-info/top_level.txt,sha256=FozpD3vmZ4WP01hjSDUt5ENn0ttnIxxm1tPrTiIsyzI,11
-traia_iatp-0.1.11.dist-info/RECORD,,
+traia_iatp-0.1.13.dist-info/licenses/LICENSE,sha256=R8vcu8GZdTTXF40sbOll11-FbarDm7_PiuaGzQp9whw,1065
+traia_iatp-0.1.13.dist-info/METADATA,sha256=NMIELzsA3ZSlVfNJXv9e8z1QjA3SieHOxEL8skCs-AE,13412
+traia_iatp-0.1.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+traia_iatp-0.1.13.dist-info/entry_points.txt,sha256=K4p9Z2lBdPEybzXfed3JbUtrVHDdg60CRnJ178df7K0,55
+traia_iatp-0.1.13.dist-info/top_level.txt,sha256=FozpD3vmZ4WP01hjSDUt5ENn0ttnIxxm1tPrTiIsyzI,11
+traia_iatp-0.1.13.dist-info/RECORD,,