prompt-caller 0.1.2__tar.gz → 0.1.3__tar.gz

{prompt_caller-0.1.2 → prompt_caller-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.1.2
+Version: 0.1.3
 Summary: This package is responsible for calling prompts in a specific format. It uses LangChain and OpenAI API
 Home-page: https://github.com/ThiNepo/prompt-caller
 Author: Thiago Nepomuceno

{prompt_caller-0.1.2 → prompt_caller-0.1.3}/prompt_caller/prompt_caller.py

@@ -8,6 +8,7 @@ from jinja2 import Template
 from langchain_core.tools import tool
 from langchain_core.messages import HumanMessage, SystemMessage, ToolMessage
 from langchain_openai import ChatOpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 from PIL import Image
 from pydantic import BaseModel, Field, create_model
 
@@ -50,6 +51,14 @@ class PromptCaller:
 
         return elements
 
+    def _createChat(self, configuration):
+        if configuration.get("model") is not None and configuration.get(
+            "model"
+        ).startswith("gemini"):
+            return ChatGoogleGenerativeAI(**configuration)
+        else:
+            return ChatOpenAI(**configuration)
+
     def getImageBase64(self, url: str) -> str:
         response = requests.get(url)
         response.raise_for_status()
@@ -119,7 +128,7 @@ class PromptCaller:
             output = configuration.get("output")
             configuration.pop("output")
 
-        chat = ChatOpenAI(**configuration)
+        chat = self._createChat(configuration)
 
         if output:
             dynamicModel = self.createPydanticModel(output)
@@ -146,7 +155,7 @@ class PromptCaller:
                     message.content += "\nOnly use the tool DynamicModel when providing an output call."
                     break
 
-        chat = ChatOpenAI(**configuration)
+        chat = self._createChat(configuration)
 
         # Register the tools
         if tools is None:
@@ -161,7 +170,7 @@ class PromptCaller:
             tools.extend([output])
             tools_dict[output.__name__.lower()] = output
         elif dynamicOutput:
-            dynamicModel = self.createPydanticModel(output)
+            dynamicModel = self.createPydanticModel(dynamicOutput)
 
             tools.extend([dynamicModel])
             tools_dict["dynamicmodel"] = dynamicModel

{prompt_caller-0.1.2 → prompt_caller-0.1.3}/prompt_caller.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.1.2
+Version: 0.1.3
 Summary: This package is responsible for calling prompts in a specific format. It uses LangChain and OpenAI API
 Home-page: https://github.com/ThiNepo/prompt-caller
 Author: Thiago Nepomuceno

{prompt_caller-0.1.2 → prompt_caller-0.1.3}/setup.py

@@ -35,7 +35,7 @@ class BdistWheelCommand(bdist_wheel):
 
 setuptools.setup(
     name="prompt_caller",
-    version="0.1.2",
+    version="0.1.3",
     author="Thiago Nepomuceno",
     author_email="thiago@neps.academy",
     description="This package is responsible for calling prompts in a specific format. It uses LangChain and OpenAI API",