Skip to content

Commit e57e53a

Browse files
sararobcopybara-github
authored andcommitted
feat: GenAI client - Add delete_file and delete_corpus methods to RAG module
PiperOrigin-RevId: 926720989
1 parent 34a2650 commit e57e53a

8 files changed

Lines changed: 1435 additions & 449 deletions

File tree

agentplatform/_genai/rag.py

Lines changed: 1160 additions & 190 deletions
Large diffs are not rendered by default.

agentplatform/_genai/types/__init__.py

Lines changed: 27 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -2527,15 +2527,24 @@
25272527
"AskContextsResponse",
25282528
"AskContextsResponseDict",
25292529
"AskContextsResponseOrDict",
2530-
"ApiAuthApiKeyConfig",
2531-
"ApiAuthApiKeyConfigDict",
2532-
"ApiAuthApiKeyConfigOrDict",
2533-
"ApiAuth",
2534-
"ApiAuthDict",
2535-
"ApiAuthOrDict",
2536-
"RagVectorDbConfigPinecone",
2537-
"RagVectorDbConfigPineconeDict",
2538-
"RagVectorDbConfigPineconeOrDict",
2530+
"CorpusStatus",
2531+
"CorpusStatusDict",
2532+
"CorpusStatusOrDict",
2533+
"RagCorpusCorpusTypeConfigDocumentCorpus",
2534+
"RagCorpusCorpusTypeConfigDocumentCorpusDict",
2535+
"RagCorpusCorpusTypeConfigDocumentCorpusOrDict",
2536+
"RagFileParsingConfigLlmParser",
2537+
"RagFileParsingConfigLlmParserDict",
2538+
"RagFileParsingConfigLlmParserOrDict",
2539+
"RagCorpusCorpusTypeConfigMemoryCorpus",
2540+
"RagCorpusCorpusTypeConfigMemoryCorpusDict",
2541+
"RagCorpusCorpusTypeConfigMemoryCorpusOrDict",
2542+
"RagCorpusCorpusTypeConfig",
2543+
"RagCorpusCorpusTypeConfigDict",
2544+
"RagCorpusCorpusTypeConfigOrDict",
2545+
"EncryptionSpec",
2546+
"EncryptionSpecDict",
2547+
"EncryptionSpecOrDict",
25392548
"RagEmbeddingModelConfigVertexPredictionEndpoint",
25402549
"RagEmbeddingModelConfigVertexPredictionEndpointDict",
25412550
"RagEmbeddingModelConfigVertexPredictionEndpointOrDict",
@@ -2551,6 +2560,15 @@
25512560
"RagEmbeddingModelConfig",
25522561
"RagEmbeddingModelConfigDict",
25532562
"RagEmbeddingModelConfigOrDict",
2563+
"ApiAuthApiKeyConfig",
2564+
"ApiAuthApiKeyConfigDict",
2565+
"ApiAuthApiKeyConfigOrDict",
2566+
"ApiAuth",
2567+
"ApiAuthDict",
2568+
"ApiAuthOrDict",
2569+
"RagVectorDbConfigPinecone",
2570+
"RagVectorDbConfigPineconeDict",
2571+
"RagVectorDbConfigPineconeOrDict",
25542572
"RagVectorDbConfigRagManagedDbANN",
25552573
"RagVectorDbConfigRagManagedDbANNDict",
25562574
"RagVectorDbConfigRagManagedDbANNOrDict",
@@ -2575,24 +2593,6 @@
25752593
"RagVectorDbConfig",
25762594
"RagVectorDbConfigDict",
25772595
"RagVectorDbConfigOrDict",
2578-
"CorpusStatus",
2579-
"CorpusStatusDict",
2580-
"CorpusStatusOrDict",
2581-
"RagCorpusCorpusTypeConfigDocumentCorpus",
2582-
"RagCorpusCorpusTypeConfigDocumentCorpusDict",
2583-
"RagCorpusCorpusTypeConfigDocumentCorpusOrDict",
2584-
"RagFileParsingConfigLlmParser",
2585-
"RagFileParsingConfigLlmParserDict",
2586-
"RagFileParsingConfigLlmParserOrDict",
2587-
"RagCorpusCorpusTypeConfigMemoryCorpus",
2588-
"RagCorpusCorpusTypeConfigMemoryCorpusDict",
2589-
"RagCorpusCorpusTypeConfigMemoryCorpusOrDict",
2590-
"RagCorpusCorpusTypeConfig",
2591-
"RagCorpusCorpusTypeConfigDict",
2592-
"RagCorpusCorpusTypeConfigOrDict",
2593-
"EncryptionSpec",
2594-
"EncryptionSpecDict",
2595-
"EncryptionSpecOrDict",
25962596
"VertexAiSearchConfig",
25972597
"VertexAiSearchConfigDict",
25982598
"VertexAiSearchConfigOrDict",

0 commit comments

Comments
 (0)