puli-plg 0.1.26__py3-none-any.whl

puli_models/incidents.py

@@ -0,0 +1,78 @@
+from dataclasses import dataclass, field
+from typing import List
+
+
+@dataclass
+class Incident:
+    """Represents an incident record in the Zilliz collection."""
+
+    id: str
+    vector: List[float]
+    company: str
+    year: int
+    industries: List[str] = field(default_factory=list)
+    tags: List[str] = field(default_factory=list)
+    trigger_keywords: List[str] = field(default_factory=list)
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "Incident":
+        """Create an Incident from a dictionary, handling Milvus/Zilliz types."""
+        return cls(
+            id=data.get("id"),
+            vector=list(data.get("vector", [])),
+            company=data.get("company"),
+            year=data.get("year"),
+            industries=list(data.get("industries", [])),
+            tags=list(data.get("tags", [])),
+            trigger_keywords=list(data.get("trigger_keywords", [])),
+        )
+
+    def to_dict(self) -> dict:
+        """Convert to dictionary for insertion."""
+        return {
+            "id": self.id,
+            "vector": self.vector,
+            "company": self.company,
+            "year": self.year,
+            "industries": self.industries,
+            "tags": self.tags,
+            "trigger_keywords": self.trigger_keywords,
+        }
+
+
+@dataclass
+class IncidentQueryResult:
+    """Represents a lightweight incident record for search/query results."""
+
+    id: str
+    company: str
+    year: int
+    industries: List[str] = field(default_factory=list)
+    tags: List[str] = field(default_factory=list)
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "IncidentQueryResult":
+        """Create a QueryResult from a dictionary, handling Milvus/Zilliz types."""
+        return cls(
+            id=data.get("id"),
+            company=data.get("company"),
+            year=data.get("year"),
+            industries=list(data.get("industries", [])),
+            tags=list(data.get("tags", [])),
+        )
+
+    def to_dict(self) -> dict:
+        """Convert to dictionary for serialization."""
+        return {
+            "id": self.id,
+            "company": self.company,
+            "year": self.year,
+            "industries": self.industries,
+            "tags": self.tags,
+        }
+
+    def to_prompt_str(self) -> str:
+        """Returns a string representation of the incident query result."""
+        return f"Incident: {self.company} ({self.year}) " + '\n' \
+            + f"Tags: {', '.join(self.tags)}" + '\n' \
+            + f"Industries: {', '.join(self.industries)}" + '\n'

puli_plg-0.1.26.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: puli-plg
+Version: 0.1.26
+Summary: Zilliz vector database client for incident management
+Requires-Python: <3.14,>=3.10
+Requires-Dist: google-auth>=2.0.0
+Requires-Dist: mcp[cli]>=0.1.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pydantic-ai>=0.0.18
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pyyaml>=6.0.1
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.1.0; extra == 'dev'

puli_plg-0.1.26.dist-info/RECORD

@@ -0,0 +1,25 @@
+puli_mcp_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+puli_mcp_server/test_diff.json,sha256=Nia-fCUjE1p9TSqjyigxzQEbCue9-p33HhTmcF6dV1s,1351
+puli_mcp_server/credentials/.gitkeep,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+puli_mcp_server/embedding_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+puli_mcp_server/embedding_client/client.py,sha256=13aNY9RteC9oNr0tPNC3JTCvQbj9hITpBxUBk7J1NHY,1047
+puli_mcp_server/embedding_client/config.py,sha256=LDrUpRQO-OYhYhiecL5LNyQ_Xh0uFRmf0t1PCh8ZOZw,1438
+puli_mcp_server/llm_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+puli_mcp_server/llm_agent/config.py,sha256=UCl1UQrVaQfnc26fmV0wSYthWE3CwZ3dY2DA5OWSMNE,2985
+puli_mcp_server/llm_agent/llm_agent.py,sha256=EC4K3uMHrYnEX3hxEmCIqyBK6j5Ga9VZleja8-dJEvQ,1449
+puli_mcp_server/llm_agent/models.py,sha256=gL2JVI4FtiuaEFqFp8jSg7FyfOL44mlH5HWaNDXi5iQ,10868
+puli_mcp_server/mcp_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+puli_mcp_server/mcp_server/models.py,sha256=TI7Qnhfb1Tt6F_5Mg_vgd6mIKLhfGYXWm9oRCiLO7-k,2302
+puli_mcp_server/mcp_server/server.py,sha256=pnT3zq11jdxjqeKv0JDwZPB6NA7Yg8CCUkwSIq6b-14,4307
+puli_mcp_server/proxy_client/__init__.py,sha256=-D9PvJPzK0AZrQWevsGe1BCQVcmcQk7P_81cpNARJf8,59
+puli_mcp_server/proxy_client/client.py,sha256=QO1hSHqxRTEvFupRYtgvIr9abTQD7XObs9uI63o9M2M,8901
+puli_mcp_server/proxy_client/config.py,sha256=QzB47oCF9KhVhkfLagNKz35CTcsjbgNaa-YyHRg-IX8,2326
+puli_mcp_server/proxy_client/token_manager.py,sha256=li2sxk-EwHHgCM7WEnpW1pWfvNWqezOunQUNF24JmX8,796
+puli_models/__init__.py,sha256=8qgpZupjCu4nmGZJhTcnF-DnAdfloe9CtHUZwh7T-pQ,228
+puli_models/chaos_patterns.py,sha256=cwUKAFz6_G4fJn8eLO_rXe0WERErvAfwPOBLHupfLfU,2701
+puli_models/incidents.py,sha256=ysQBbG7yyLe2w6e1WhNahuohlCGp1EUzVw0LZOPXO80,2524
+puli_mcp_server/credentials/service-account.json,sha256=4Zng_2Y-YXzgwcaWJreGaumQ-nNYQymrUHx6D_Bj060,2399
+puli_plg-0.1.26.dist-info/METADATA,sha256=_e1QfSKZ741GVsIS3ZJQamGaRdLqrDif5bhbUOz6Pl0,448
+puli_plg-0.1.26.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+puli_plg-0.1.26.dist-info/entry_points.txt,sha256=FE5DJ7iMEWzWmTCGVJ8IMdr_xUIcHi-Nq99TkqOpRyo,73
+puli_plg-0.1.26.dist-info/RECORD,,

puli_plg-0.1.26.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

puli_plg-0.1.26.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+puli-reviewer = puli_mcp_server.mcp_server.server:main