genkit 0.5.1__tar.gz → 0.5.2__tar.gz

{genkit-0.5.1 → genkit-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genkit
-Version: 0.5.1
+Version: 0.5.2
 Summary: Genkit AI Framework
 Project-URL: Bug Tracker, https://github.com/firebase/genkit/issues
 Project-URL: Changelog, https://github.com/firebase/genkit/blob/main/py/packages/genkit/CHANGELOG.md
@@ -52,8 +52,6 @@ Requires-Dist: structlog>=25.2.0
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: uvicorn>=0.34.0
 Requires-Dist: uvloop>=0.21.0; sys_platform != 'win32'
-Provides-Extra: dev-local-vectorstore
-Requires-Dist: genkit-plugin-dev-local-vectorstore; extra == 'dev-local-vectorstore'
 Provides-Extra: flask
 Requires-Dist: genkit-plugin-flask; extra == 'flask'
 Provides-Extra: google-cloud
@@ -88,7 +86,7 @@ pip install genkit-plugin-google-genai
 
 ```python
 from pydantic import BaseModel, Field
-from genkit.ai import Genkit, Output
+from genkit import Genkit
 from genkit.plugins.google_genai import GoogleAI
 
 ai = Genkit(
@@ -96,6 +94,7 @@ ai = Genkit(
     model='googleai/gemini-2.0-flash',
 )
 
+
 class RpgCharacter(BaseModel):
     """An RPG game character."""
 
@@ -103,11 +102,12 @@ class RpgCharacter(BaseModel):
     back_story: str = Field(description='back story')
     abilities: list[str] = Field(description='list of abilities (3-4)')
 
+
 @ai.flow()
 async def generate_character(name: str) -> RpgCharacter:
     result = await ai.generate(
         prompt=f'generate an RPG character named {name}',
-        output=Output(schema=RpgCharacter),
+        output_schema=RpgCharacter,
     )
     return result.output
 
@@ -117,6 +117,7 @@ async def main() -> None:
     character = await generate_character('Goblorb')
     print(character.model_dump_json(indent=2))
 
+
 if __name__ == '__main__':
     ai.run_main(main())
 ```

{genkit-0.5.1 → genkit-0.5.2}/README.md

@@ -18,7 +18,7 @@ pip install genkit-plugin-google-genai
 
 ```python
 from pydantic import BaseModel, Field
-from genkit.ai import Genkit, Output
+from genkit import Genkit
 from genkit.plugins.google_genai import GoogleAI
 
 ai = Genkit(
@@ -26,6 +26,7 @@ ai = Genkit(
     model='googleai/gemini-2.0-flash',
 )
 
+
 class RpgCharacter(BaseModel):
     """An RPG game character."""
 
@@ -33,11 +34,12 @@ class RpgCharacter(BaseModel):
     back_story: str = Field(description='back story')
     abilities: list[str] = Field(description='list of abilities (3-4)')
 
+
 @ai.flow()
 async def generate_character(name: str) -> RpgCharacter:
     result = await ai.generate(
         prompt=f'generate an RPG character named {name}',
-        output=Output(schema=RpgCharacter),
+        output_schema=RpgCharacter,
     )
     return result.output
 
@@ -47,6 +49,7 @@ async def main() -> None:
     character = await generate_character('Goblorb')
     print(character.model_dump_json(indent=2))
 
+
 if __name__ == '__main__':
     ai.run_main(main())
 ```

{genkit-0.5.1 → genkit-0.5.2}/pyproject.toml

@@ -54,6 +54,7 @@ dependencies = [
   "pillow>=12.1.1",
   "typing_extensions>=4.0",
   "strenum>=0.4.15; python_version < '3.11'",
+
   "dotpromptz>=0.1.5",
   "uvicorn>=0.34.0",
   "uvloop>=0.21.0; sys_platform != 'win32'",
@@ -75,10 +76,9 @@ license = "Apache-2.0"
 name = "genkit"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.5.1"
+version = "0.5.2"
 
 [project.optional-dependencies]
-dev-local-vectorstore = ["genkit-plugin-dev-local-vectorstore"]
 flask                 = ["genkit-plugin-flask"]
 google-cloud          = ["genkit-plugin-google-cloud"]
 google-genai          = ["genkit-plugin-google-genai"]
@@ -106,7 +106,6 @@ testpaths  = ["tests"]
 
 [tool.uv.sources]
 genkit-plugin-compat-oai            = { workspace = true }
-genkit-plugin-dev-local-vectorstore = { workspace = true }
 genkit-plugin-flask                 = { workspace = true }
 genkit-plugin-google-genai          = { workspace = true }
 genkit-plugin-ollama                = { workspace = true }

genkit-0.5.2/src/genkit/__init__.py

@@ -0,0 +1,135 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+"""Genkit — Build AI-powered applications."""
+
+from genkit._ai._aio import ActionKind, ActionRunContext, Genkit
+from genkit._ai._prompt import (
+    ExecutablePrompt,
+    ModelStreamResponse,
+    PromptGenerateOptions,
+    ResumeOptions,
+)
+from genkit._ai._tools import ToolInterruptError, ToolRunContext, tool_response
+from genkit._core._action import Action, StreamResponse
+from genkit._core._error import GenkitError, PublicError
+from genkit._core._model import Document
+from genkit._core._plugin import Plugin
+from genkit._core._plugins import extend_plugin_namespace
+from genkit._core._typing import (
+    CustomPart,
+    DocumentPart,
+    Media,
+    MediaPart,
+    Metadata,
+    Part,
+    ReasoningPart,
+    Role,
+    TextPart,
+    ToolChoice,
+    ToolRequest,
+    ToolRequestPart,
+    ToolResponse,
+    ToolResponsePart,
+)
+
+# Import embedder-related types from the embedder namespace
+from genkit.embedder import (
+    EmbedderOptions,
+    EmbedderRef,
+    Embedding,
+    EmbedRequest,
+    EmbedResponse,
+)
+
+# Import model-related types from the model namespace.
+from genkit.model import (
+    Constrained,
+    FinishReason,
+    Message,
+    ModelConfig,
+    ModelInfo,
+    ModelRequest,
+    ModelResponse,
+    ModelResponseChunk,
+    ModelUsage,
+    Stage,
+    Supports,
+    ToolDefinition,
+)
+
+# Flow is an alias for Action (used in samples for flow type hints)
+Flow = Action
+
+extend_plugin_namespace()
+
+__all__ = [
+    # Main class
+    'Genkit',
+    'Flow',
+    # Response types
+    'Action',
+    'StreamResponse',
+    'EmbedRequest',
+    'EmbedResponse',
+    'EmbedderOptions',
+    'EmbedderRef',
+    'ModelConfig',
+    'ModelInfo',
+    'ModelStreamResponse',
+    # Errors
+    'GenkitError',
+    'PublicError',
+    'ToolInterruptError',
+    # Content types
+    'Constrained',
+    'CustomPart',
+    'Embedding',
+    'Metadata',
+    'ReasoningPart',
+    'FinishReason',
+    'ModelUsage',
+    'Media',
+    'MediaPart',
+    'Message',
+    'Part',
+    'Role',
+    'Stage',
+    'Supports',
+    'TextPart',
+    'ToolChoice',
+    'ToolDefinition',
+    'ToolRequest',
+    'ToolRequestPart',
+    'ToolResponse',
+    'ToolResponsePart',
+    # Domain types
+    'Document',
+    'DocumentPart',
+    # Plugin interface
+    'Plugin',
+    # AI runtime
+    'ActionKind',
+    'ActionRunContext',
+    'ExecutablePrompt',
+    'PromptGenerateOptions',
+    'ResumeOptions',
+    'ToolRunContext',
+    'tool_response',
+    'ModelRequest',
+    'ModelResponse',
+    'ModelResponseChunk',
+]