|
6 | 6 | from ..core.request_options import RequestOptions |
7 | 7 | from ..types.add_triple_response import AddTripleResponse |
8 | 8 | from ..types.clone_graph_response import CloneGraphResponse |
9 | | -from ..types.entity_type_response import EntityTypeResponse |
10 | 9 | from ..types.episode import Episode |
11 | 10 | from ..types.episode_data import EpisodeData |
12 | 11 | from ..types.fact_rating_instruction import FactRatingInstruction |
@@ -616,86 +615,6 @@ def update( |
616 | 615 | ) |
617 | 616 | return _response.data |
618 | 617 |
|
619 | | - def set_ontology( |
620 | | - self, |
621 | | - *, |
622 | | - entities: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT, |
623 | | - edges: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT, |
624 | | - user_ids: typing.Optional[typing.Sequence[str]] = OMIT, |
625 | | - graph_ids: typing.Optional[typing.Sequence[str]] = OMIT, |
626 | | - request_options: typing.Optional[RequestOptions] = None, |
627 | | - ) -> SuccessResponse: |
628 | | - """ |
629 | | - Sets custom entity and edge types for your graph. This wrapper method |
630 | | - provides a clean interface for defining your graph schema with custom |
631 | | - entity and edge types. |
632 | | -
|
633 | | - See the [full documentation](/customizing-graph-structure#setting-entity-and-edge-types) for details. |
634 | | -
|
635 | | - Parameters |
636 | | - ---------- |
637 | | - entities : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] |
638 | | - Dictionary mapping entity type names to their definitions |
639 | | -
|
640 | | - edges : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] |
641 | | - Dictionary mapping edge type names to their definitions with source/target constraints |
642 | | -
|
643 | | - user_ids : typing.Optional[typing.Sequence[str]] |
644 | | - Optional list of user IDs to apply ontology to |
645 | | -
|
646 | | - graph_ids : typing.Optional[typing.Sequence[str]] |
647 | | - Optional list of graph IDs to apply ontology to |
648 | | -
|
649 | | - request_options : typing.Optional[RequestOptions] |
650 | | - Request-specific configuration. |
651 | | -
|
652 | | - Returns |
653 | | - ------- |
654 | | - SuccessResponse |
655 | | - Ontology set successfully |
656 | | -
|
657 | | - Examples |
658 | | - -------- |
659 | | - from zep_cloud import Zep |
660 | | -
|
661 | | - client = Zep( |
662 | | - api_key="YOUR_API_KEY", |
663 | | - ) |
664 | | - client.graph.set_ontology() |
665 | | - """ |
666 | | - _response = self._raw_client.set_ontology( |
667 | | - entities=entities, edges=edges, user_ids=user_ids, graph_ids=graph_ids, request_options=request_options |
668 | | - ) |
669 | | - return _response.data |
670 | | - |
671 | | - def list_ontology(self, *, request_options: typing.Optional[RequestOptions] = None) -> EntityTypeResponse: |
672 | | - """ |
673 | | - Retrieves the current entity and edge types configured for your graph. |
674 | | -
|
675 | | - See the [full documentation](/customizing-graph-structure) for details. |
676 | | -
|
677 | | - Parameters |
678 | | - ---------- |
679 | | - request_options : typing.Optional[RequestOptions] |
680 | | - Request-specific configuration. |
681 | | -
|
682 | | - Returns |
683 | | - ------- |
684 | | - EntityTypeResponse |
685 | | - Current ontology |
686 | | -
|
687 | | - Examples |
688 | | - -------- |
689 | | - from zep_cloud import Zep |
690 | | -
|
691 | | - client = Zep( |
692 | | - api_key="YOUR_API_KEY", |
693 | | - ) |
694 | | - client.graph.list_ontology() |
695 | | - """ |
696 | | - _response = self._raw_client.list_ontology(request_options=request_options) |
697 | | - return _response.data |
698 | | - |
699 | 618 |
|
700 | 619 | class AsyncGraphClient: |
701 | 620 | def __init__(self, *, client_wrapper: AsyncClientWrapper): |
@@ -1367,99 +1286,3 @@ async def main() -> None: |
1367 | 1286 | request_options=request_options, |
1368 | 1287 | ) |
1369 | 1288 | return _response.data |
1370 | | - |
1371 | | - async def set_ontology( |
1372 | | - self, |
1373 | | - *, |
1374 | | - entities: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT, |
1375 | | - edges: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT, |
1376 | | - user_ids: typing.Optional[typing.Sequence[str]] = OMIT, |
1377 | | - graph_ids: typing.Optional[typing.Sequence[str]] = OMIT, |
1378 | | - request_options: typing.Optional[RequestOptions] = None, |
1379 | | - ) -> SuccessResponse: |
1380 | | - """ |
1381 | | - Sets custom entity and edge types for your graph. This wrapper method |
1382 | | - provides a clean interface for defining your graph schema with custom |
1383 | | - entity and edge types. |
1384 | | -
|
1385 | | - See the [full documentation](/customizing-graph-structure#setting-entity-and-edge-types) for details. |
1386 | | -
|
1387 | | - Parameters |
1388 | | - ---------- |
1389 | | - entities : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] |
1390 | | - Dictionary mapping entity type names to their definitions |
1391 | | -
|
1392 | | - edges : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] |
1393 | | - Dictionary mapping edge type names to their definitions with source/target constraints |
1394 | | -
|
1395 | | - user_ids : typing.Optional[typing.Sequence[str]] |
1396 | | - Optional list of user IDs to apply ontology to |
1397 | | -
|
1398 | | - graph_ids : typing.Optional[typing.Sequence[str]] |
1399 | | - Optional list of graph IDs to apply ontology to |
1400 | | -
|
1401 | | - request_options : typing.Optional[RequestOptions] |
1402 | | - Request-specific configuration. |
1403 | | -
|
1404 | | - Returns |
1405 | | - ------- |
1406 | | - SuccessResponse |
1407 | | - Ontology set successfully |
1408 | | -
|
1409 | | - Examples |
1410 | | - -------- |
1411 | | - import asyncio |
1412 | | -
|
1413 | | - from zep_cloud import AsyncZep |
1414 | | -
|
1415 | | - client = AsyncZep( |
1416 | | - api_key="YOUR_API_KEY", |
1417 | | - ) |
1418 | | -
|
1419 | | -
|
1420 | | - async def main() -> None: |
1421 | | - await client.graph.set_ontology() |
1422 | | -
|
1423 | | -
|
1424 | | - asyncio.run(main()) |
1425 | | - """ |
1426 | | - _response = await self._raw_client.set_ontology( |
1427 | | - entities=entities, edges=edges, user_ids=user_ids, graph_ids=graph_ids, request_options=request_options |
1428 | | - ) |
1429 | | - return _response.data |
1430 | | - |
1431 | | - async def list_ontology(self, *, request_options: typing.Optional[RequestOptions] = None) -> EntityTypeResponse: |
1432 | | - """ |
1433 | | - Retrieves the current entity and edge types configured for your graph. |
1434 | | -
|
1435 | | - See the [full documentation](/customizing-graph-structure) for details. |
1436 | | -
|
1437 | | - Parameters |
1438 | | - ---------- |
1439 | | - request_options : typing.Optional[RequestOptions] |
1440 | | - Request-specific configuration. |
1441 | | -
|
1442 | | - Returns |
1443 | | - ------- |
1444 | | - EntityTypeResponse |
1445 | | - Current ontology |
1446 | | -
|
1447 | | - Examples |
1448 | | - -------- |
1449 | | - import asyncio |
1450 | | -
|
1451 | | - from zep_cloud import AsyncZep |
1452 | | -
|
1453 | | - client = AsyncZep( |
1454 | | - api_key="YOUR_API_KEY", |
1455 | | - ) |
1456 | | -
|
1457 | | -
|
1458 | | - async def main() -> None: |
1459 | | - await client.graph.list_ontology() |
1460 | | -
|
1461 | | -
|
1462 | | - asyncio.run(main()) |
1463 | | - """ |
1464 | | - _response = await self._raw_client.list_ontology(request_options=request_options) |
1465 | | - return _response.data |
0 commit comments