llumo 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

llumo/functionCalling.py

@@ -2,19 +2,19 @@ import os
 import json
 import pandas as pd
 from typing import Callable, List, Dict
-from dotenv import load_dotenv
 import google.generativeai as genai
 from google.generativeai.types import FunctionDeclaration, Tool
 from openai import OpenAI
 from .exceptions import *
-# Load environment variables
-load_dotenv()
+
 # openai_api = os.getenv("OPENAI_API_KEY")
 # google_api_key = os.getenv("GOOGLE_API_KEY")
 
 
 class LlumoAgent:
-    def __init__(self, name: str, description: str, parameters: Dict, function: Callable):
+    def __init__(
+        self, name: str, description: str, parameters: Dict, function: Callable
+    ):
         self.name = name
         self.description = description
         self.parameters = parameters
@@ -30,8 +30,8 @@ class LlumoAgent:
             parameters={
                 "type": "object",
                 "properties": self.parameters,
-                "required": list(self.parameters.keys())
-            }
+                "required": list(self.parameters.keys()),
+            },
         )
 
     def createOpenaiTool(self):
@@ -44,25 +44,28 @@ class LlumoAgent:
                     "type": "object",
                     "properties": self.parameters,
                     "required": list(self.parameters.keys()),
-                }
-            }
+                },
+            },
         }
 
 
 class LlumoAgentExecutor:
 
     @staticmethod
-    def run(df: pd.DataFrame, agents: List[LlumoAgent], model: str,model_api_key = None):
+    def run(df: pd.DataFrame, agents: List[LlumoAgent], model: str, model_api_key=None):
         if model.lower() == "google":
-            return LlumoAgentExecutor.runWithGoogle(df, agents,model_api_key = model_api_key)
+            return LlumoAgentExecutor.runWithGoogle(
+                df, agents, model_api_key=model_api_key
+            )
         elif model.lower() == "openai":
-            return LlumoAgentExecutor.runWithOpenAI(df, agents,model_api_key = model_api_key)
+            return LlumoAgentExecutor.runWithOpenAI(
+                df, agents, model_api_key=model_api_key
+            )
         else:
             raise ValueError(f"Unsupported model: {model}. Use 'google' or 'openai'.")
 
-
     @staticmethod
-    def runWithGoogle(df: pd.DataFrame, agents: List[LlumoAgent],model_api_key = None):
+    def runWithGoogle(df: pd.DataFrame, agents: List[LlumoAgent], model_api_key=None):
         try:
             genai.configure(api_key=model_api_key)
             tool_defs = []
@@ -94,7 +97,7 @@ class LlumoAgentExecutor:
                             follow_up_msg = {
                                 "function_response": {
                                     "name": func_call.name,
-                                    "response": result
+                                    "response": result,
                                 }
                             }
 
@@ -119,7 +122,7 @@ class LlumoAgentExecutor:
             raise RuntimeError(f"Error in runWithGoogle: {e}")
 
     @staticmethod
-    def runWithOpenAI(df: pd.DataFrame, agents: List[LlumoAgent],model_api_key = None):
+    def runWithOpenAI(df: pd.DataFrame, agents: List[LlumoAgent], model_api_key=None):
         try:
             client = OpenAI(api_key=model_api_key)
             all_tools = [agent.createOpenaiTool() for agent in agents]
@@ -137,7 +140,7 @@ class LlumoAgentExecutor:
                         model="gpt-4",
                         messages=messages,
                         tools=all_tools,
-                        tool_choice="auto"
+                        tool_choice="auto",
                     )
 
                     first_message = initial_response.choices[0].message
@@ -157,15 +160,16 @@ class LlumoAgentExecutor:
                             agent = agent_lookup[tool_name]
                             tool_result = agent.run(**args)
 
-                            messages.append({
-                                "role": "tool",
-                                "tool_call_id": call.id,
-                                "content": str(tool_result)
-                            })
+                            messages.append(
+                                {
+                                    "role": "tool",
+                                    "tool_call_id": call.id,
+                                    "content": str(tool_result),
+                                }
+                            )
 
                         final_response = client.chat.completions.create(
-                            model="gpt-4",
-                            messages=messages
+                            model="gpt-4", messages=messages
                         )
                         final_output = final_response.choices[0].message.content
                         messages.append({"role": "assistant", "content": final_output})
@@ -187,4 +191,3 @@ class LlumoAgentExecutor:
 
         except Exception as e:
             raise RuntimeError(f"Error in runWithOpenAI: {e}")
-

llumo/helpingFuntions.py

@@ -7,10 +7,6 @@ import requests
 import json
 import base64
 import os
-from dotenv import load_dotenv
-
-
-load_dotenv()
 
 subscriptionUrl = "https://app.llumo.ai/api/workspace/record-extra-usage"
 getStreamdataUrl = "https://app.llumo.ai/api/data-stream/all"

{llumo-0.2.3.dist-info → llumo-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llumo
-Version: 0.2.3
+Version: 0.2.4
 Summary: Python SDK for interacting with the Llumo ai API.
 Home-page: https://www.llumo.ai/
 Author: Llumo

{llumo-0.2.3.dist-info → llumo-0.2.4.dist-info}/RECORD

@@ -2,12 +2,12 @@ llumo/__init__.py,sha256=O04b4yW1BnOvcHzxWFddAKhtdBEhBNhLdb6xgnpHH_Q,205
 llumo/client.py,sha256=aCYeUrmEssSqydA_p3nokJ9MkspgitJ0HTmzL7SpkQw,35540
 llumo/exceptions.py,sha256=iCj7HhtO_ckC2EaVBdXbAudNpuMDsYmmMEV5lwynZ-E,1854
 llumo/execution.py,sha256=x88wQV8eL99wNN5YtjFaAMCIfN1PdfQVlAZQb4vzgQ0,1413
-llumo/functionCalling.py,sha256=QtuTtyoz5rnfNUrNT1kzegNPOrMFjrlgxZfwTqRMdiA,7190
-llumo/helpingFuntions.py,sha256=mDGOjeqJI1NOdoaT5FK3tlhWwhpZnk1nCnZOan0AFk0,8886
+llumo/functionCalling.py,sha256=D5jYapu1rIvdIJNUYPYMTyhQ1H-6nkwoOLMi6eekfUE,7241
+llumo/helpingFuntions.py,sha256=G_pqLhYNH3bZ47gA--w6YFNvB443xkrbh8CeQfYxazk,8839
 llumo/models.py,sha256=YH-qAMnShmUpmKE2LQAzQdpRsaXkFSlOqMxHwU4zBUI,1560
 llumo/sockets.py,sha256=Qxxqtx3Hg07HLhA4QfcipK1ChiOYhHZBu02iA6MfYlQ,5579
-llumo-0.2.3.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
-llumo-0.2.3.dist-info/METADATA,sha256=_Iv2gg-JKGBJvL8nEELMGJNxNUi8fBWqggI8Fas0gOY,426
-llumo-0.2.3.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-llumo-0.2.3.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
-llumo-0.2.3.dist-info/RECORD,,
+llumo-0.2.4.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
+llumo-0.2.4.dist-info/METADATA,sha256=fccBu5Rs8SdsH-PstSiI_jeAW7Be1LR5DnG6quUdclc,426
+llumo-0.2.4.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+llumo-0.2.4.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.2.4.dist-info/RECORD,,