npcsh 0.3.26__tar.gz → 0.3.27__tar.gz

{npcsh-0.3.26/npcsh.egg-info → npcsh-0.3.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: npcsh
-Version: 0.3.26
+Version: 0.3.27
 Summary: npcsh is a command line tool for integrating LLMs into everyday workflows and for orchestrating teams of NPCs.
 Home-page: https://github.com/cagostino/npcsh
 Author: Christopher Agostino
@@ -13,34 +13,18 @@ License-File: LICENSE
 Requires-Dist: redis
 Requires-Dist: flask_sse
 Requires-Dist: anthropic
-Requires-Dist: screeninfo
-Requires-Dist: sentence_transformers
-Requires-Dist: nltk
-Requires-Dist: thefuzz
 Requires-Dist: beautifulsoup4
 Requires-Dist: google-generativeai
 Requires-Dist: google-genai
 Requires-Dist: duckduckgo-search
-Requires-Dist: pypdf
-Requires-Dist: PyMuPDF
-Requires-Dist: opencv-python
-Requires-Dist: librosa
 Requires-Dist: openai
 Requires-Dist: jinja2
-Requires-Dist: pandas
-Requires-Dist: matplotlib
-Requires-Dist: IPython
-Requires-Dist: ollama
 Requires-Dist: requests
 Requires-Dist: markdown
 Requires-Dist: PyYAML
 Requires-Dist: langchain
 Requires-Dist: langchain_community
-Requires-Dist: pyautogui
 Requires-Dist: pygments
-Requires-Dist: pyttsx3
-Requires-Dist: kuzu
-Requires-Dist: chromadb
 Requires-Dist: termcolor
 Requires-Dist: colorama
 Requires-Dist: python-dotenv
@@ -48,6 +32,21 @@ Requires-Dist: pytest
 Requires-Dist: googlesearch-python
 Requires-Dist: flask
 Requires-Dist: flask_cors
+Requires-Dist: librosa
+Requires-Dist: pandas
+Requires-Dist: matplotlib
+Requires-Dist: IPython
+Requires-Dist: pyautogui
+Requires-Dist: nltk
+Requires-Dist: thefuzz
+Requires-Dist: pypdf
+Requires-Dist: PyMuPDF
+Requires-Dist: screeninfo
+Requires-Dist: sentence_transformers
+Requires-Dist: opencv-python
+Requires-Dist: ollama
+Requires-Dist: kuzu
+Requires-Dist: chromadb
 Requires-Dist: diffusers
 Dynamic: author
 Dynamic: author-email

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/conversation.py

@@ -7,9 +7,8 @@
 from typing import Any, Dict, Generator, List
 import os
 import anthropic
-import ollama  # Add to setup.py if missing
+
 from openai import OpenAI
-from diffusers import StableDiffusionPipeline
 from google.generativeai import types
 import google.generativeai as genai
 from .npc_sysenv import get_system_message
@@ -34,6 +33,7 @@ def get_ollama_conversation(
     Returns:
         List[Dict[str, str]]: The list of messages in the conversation.
     """
+    import ollama
 
     messages_copy = messages.copy()
     if messages_copy[0]["role"] != "system":

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/embeddings.py

@@ -12,7 +12,6 @@ from npcsh.npc_sysenv import (
     NPCSH_EMBEDDING_PROVIDER,
     chroma_client,
 )
-import ollama
 from openai import OpenAI
 import anthropic
 
@@ -21,6 +20,8 @@ def get_ollama_embeddings(
     texts: List[str], model: str = "nomic-embed-text"
 ) -> List[List[float]]:
     """Generate embeddings using Ollama."""
+    import ollama
+
     embeddings = []
     for text in texts:
         response = ollama.embeddings(model=model, prompt=text)

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/image_gen.py

@@ -10,7 +10,6 @@
 import os
 
 from openai import OpenAI
-from diffusers import StableDiffusionPipeline
 
 
 def generate_image_openai(
@@ -66,6 +65,8 @@ def generate_image_hf_diffusion(
         PIL.Image: The generated image.
     """
     # Load the Stable Diffusion pipeline
+    from diffusers import StableDiffusionPipeline
+
     pipe = StableDiffusionPipeline.from_pretrained(model)
     pipe = pipe.to(device)
 

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/knowledge_graph.py

@@ -3,7 +3,11 @@ import os
 import datetime
 
 import numpy as np
-import kuzu
+
+try:
+    import kuzu
+except ModuleNotFoundError:
+    print("kuzu not installed")
 from typing import Optional, Dict, List, Union, Tuple
 
 
@@ -581,7 +585,10 @@ def visualize_graph(conn):
     plt.show()
 
 
-import chromadb
+try:
+    import chromadb
+except ModuleNotFoundError:
+    print("chromadb not installed")
 import numpy as np
 import os
 import datetime

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/llm_funcs.py

@@ -4,17 +4,10 @@ import requests
 import os
 import json
 import PIL
-from PIL import Image
 
 import sqlite3
 from datetime import datetime
 from typing import List, Dict, Any, Optional, Union, Generator
-import typing_extensions as typing
-from pydantic import BaseModel, Field
-
-import base64
-import re
-import io
 
 
 from jinja2 import Environment, FileSystemLoader, Template, Undefined
@@ -22,15 +15,6 @@ from jinja2 import Environment, FileSystemLoader, Template, Undefined
 import pandas as pd
 import numpy as np
 
-# chroma
-import chromadb
-from chromadb import Client
-
-# llm providers
-import anthropic
-import ollama  # Add to setup.py if missing
-from openai import OpenAI
-from diffusers import StableDiffusionPipeline
 from google.generativeai import types
 import google.generativeai as genai
 
@@ -54,7 +38,6 @@ from .npc_sysenv import (
     NPCSH_API_URL,
     NPCSH_VISION_MODEL,
     NPCSH_VISION_PROVIDER,
-    chroma_client,
     available_reasoning_models,
     available_chat_models,
 )
@@ -150,25 +133,30 @@ def generate_image(
     #    image = generate_image_openai_like(prompt, model, npc.api_url, openai_api_key)
     elif provider == "diffusers":
         image = generate_image_hf_diffusion(prompt, model)
+    else:
+        image = None
     # save image
     # check if image is a PIL image
     if isinstance(image, PIL.Image.Image):
         image.save(filename)
         return filename
 
-    elif image is not None:
-        # image is at a private url
-        response = requests.get(image.data[0].url)
-        with open(filename, "wb") as file:
-            file.write(response.content)
-        from PIL import Image
+    else:
+        try:
+            # image is at a private url
+            response = requests.get(image.data[0].url)
+            with open(filename, "wb") as file:
+                file.write(response.content)
+            from PIL import Image
 
-        img = Image.open(filename)
-        img.show()
-        # console = Console()
-        # console.print(Image.from_path(filename))
+            img = Image.open(filename)
+            img.show()
+            # console = Console()
+            # console.print(Image.from_path(filename))
+            return filename
 
-        return filename
+        except AttributeError as e:
+            print(f"Error saving image: {e}")
 
 
 def get_embeddings(
@@ -511,7 +499,6 @@ def execute_llm_question(
         # messages.append({"role": "assistant", "content": output})
 
     else:
-
         response = get_conversation(
             messages,
             model=model,
@@ -1030,7 +1017,6 @@ ReAct choices then will enter reasoning flow
         return {"messages": messages, "output": output}
 
     elif action == "answer_question":
-
         if ENTER_REASONING_FLOW:
             print("entering reasoning flow")
             result = enter_reasoning_human_in_the_loop(
@@ -1208,7 +1194,6 @@ def handle_tool_call(
     # print(npc)
     print("handling tool call")
     if not npc:
-
         print(
             f"No tools available for NPC '{npc.name}' or tools_dict is empty. Available tools: {available_tools}"
         )
@@ -1320,7 +1305,6 @@ def handle_tool_call(
     print("Executing tool with input values:", input_values)
 
     try:
-
         tool_output = tool.execute(
             input_values,
             npc.all_tools_dict,
@@ -1335,7 +1319,6 @@ def handle_tool_call(
         if "Error" in tool_output:
             raise Exception(tool_output)
     except Exception as e:
-
         # diagnose_problem = get_llm_response(
         ##    f"""a problem has occurred.
         #                                    Please provide a diagnosis of the problem and a suggested #fix.

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/npc_sysenv.py

@@ -3,7 +3,6 @@ from datetime import datetime
 from typing import Any
 import os
 import io
-import chromadb
 import sqlite3
 from dotenv import load_dotenv
 from PIL import Image
@@ -222,7 +221,12 @@ available_chat_models, available_reasoning_models = get_available_models()
 
 EMBEDDINGS_DB_PATH = os.path.expanduser("~/npcsh_chroma.db")
 
-chroma_client = chromadb.PersistentClient(path=EMBEDDINGS_DB_PATH)
+try:
+    import chromadb
+
+    chroma_client = chromadb.PersistentClient(path=EMBEDDINGS_DB_PATH)
+except:
+    chroma_client = None
 
 
 # Load environment variables from .env file

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/response.py

@@ -2,9 +2,7 @@ from typing import Any, Dict, Generator, List, Union
 from pydantic import BaseModel
 import os
 import anthropic
-import ollama  # Add to setup.py if missing
 from openai import OpenAI
-from diffusers import StableDiffusionPipeline
 from google.generativeai import types
 from google import genai
 
@@ -143,6 +141,8 @@ def get_ollama_response(
     Returns:
         Dict[str, Any]: The response, optionally including updated messages.
     """
+    import ollama
+
     # try:
     # Prepare the message payload
     system_message = get_system_message(npc) if npc else "You are a helpful assistant."

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/search.py

@@ -162,8 +162,12 @@ def rag_search(
 
     """
     if embedding_model is None:
-        embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
-
+        try:
+            embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
+        except:
+            raise Exception(
+                "Please install the sentence-transformers library to use this function or provide an embedding transformer model."
+            )
     results = []
 
     # Compute the embedding of the query

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/shell_helpers.py

@@ -1433,7 +1433,6 @@ def execute_slash_command(
             command_parts, model=model, provider=provider, npc=npc, api_url=api_url
         )
     elif command_name == "help":  # New help command
-
         return {
             "messages": messages,
             "output": get_help(),
@@ -1763,7 +1762,7 @@ def execute_command(
     messages: list = None,
     conversation_id: str = None,
     stream: bool = False,
-    embedding_model: Union[SentenceTransformer, Any] = None,
+    embedding_model=None,
 ):
     """
     Function Description:
@@ -1774,7 +1773,7 @@ def execute_command(
         db_path : str : Database path
         npc_compiler : NPCCompiler : NPC compiler
     Keyword Args:
-        embedding_model : Union[SentenceTransformer, Any] : Embedding model
+        embedding_model :  Embedding model
         current_npc : NPC : Current NPC
         messages : list : Messages
     Returns:
@@ -2086,7 +2085,7 @@ def execute_command_stream(
     command: str,
     db_path: str,
     npc_compiler: NPCCompiler,
-    embedding_model: Union[SentenceTransformer, Any] = None,
+    embedding_model=None,
     current_npc: NPC = None,
     model: str = None,
     provider: str = None,

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh/stream.py

@@ -10,9 +10,7 @@ from npcsh.npc_sysenv import get_system_message
 from typing import Any, Dict, Generator, List
 import os
 import anthropic
-import ollama  # Add to setup.py if missing
 from openai import OpenAI
-from diffusers import StableDiffusionPipeline
 from google import genai
 
 from google.generativeai import types
@@ -53,6 +51,8 @@ def get_anthropic_stream(
         messages = messages[1:]
     elif npc is not None:
         system_message = get_system_message(npc)
+    else:
+        system_message = "You are a helpful assistant."
 
     # Preprocess messages to ensure content is a list of dicts
     for message in messages:
@@ -274,6 +274,8 @@ def get_ollama_stream(
     **kwargs,
 ) -> Generator:
     """Streams responses from Ollama, supporting images and tools."""
+    import ollama
+
     messages_copy = messages.copy()
 
     # Handle images if provided

{npcsh-0.3.26 → npcsh-0.3.27/npcsh.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: npcsh
-Version: 0.3.26
+Version: 0.3.27
 Summary: npcsh is a command line tool for integrating LLMs into everyday workflows and for orchestrating teams of NPCs.
 Home-page: https://github.com/cagostino/npcsh
 Author: Christopher Agostino
@@ -13,34 +13,18 @@ License-File: LICENSE
 Requires-Dist: redis
 Requires-Dist: flask_sse
 Requires-Dist: anthropic
-Requires-Dist: screeninfo
-Requires-Dist: sentence_transformers
-Requires-Dist: nltk
-Requires-Dist: thefuzz
 Requires-Dist: beautifulsoup4
 Requires-Dist: google-generativeai
 Requires-Dist: google-genai
 Requires-Dist: duckduckgo-search
-Requires-Dist: pypdf
-Requires-Dist: PyMuPDF
-Requires-Dist: opencv-python
-Requires-Dist: librosa
 Requires-Dist: openai
 Requires-Dist: jinja2
-Requires-Dist: pandas
-Requires-Dist: matplotlib
-Requires-Dist: IPython
-Requires-Dist: ollama
 Requires-Dist: requests
 Requires-Dist: markdown
 Requires-Dist: PyYAML
 Requires-Dist: langchain
 Requires-Dist: langchain_community
-Requires-Dist: pyautogui
 Requires-Dist: pygments
-Requires-Dist: pyttsx3
-Requires-Dist: kuzu
-Requires-Dist: chromadb
 Requires-Dist: termcolor
 Requires-Dist: colorama
 Requires-Dist: python-dotenv
@@ -48,6 +32,21 @@ Requires-Dist: pytest
 Requires-Dist: googlesearch-python
 Requires-Dist: flask
 Requires-Dist: flask_cors
+Requires-Dist: librosa
+Requires-Dist: pandas
+Requires-Dist: matplotlib
+Requires-Dist: IPython
+Requires-Dist: pyautogui
+Requires-Dist: nltk
+Requires-Dist: thefuzz
+Requires-Dist: pypdf
+Requires-Dist: PyMuPDF
+Requires-Dist: screeninfo
+Requires-Dist: sentence_transformers
+Requires-Dist: opencv-python
+Requires-Dist: ollama
+Requires-Dist: kuzu
+Requires-Dist: chromadb
 Requires-Dist: diffusers
 Dynamic: author
 Dynamic: author-email

{npcsh-0.3.26 → npcsh-0.3.27}/npcsh.egg-info/requires.txt

@@ -1,34 +1,18 @@
 redis
 flask_sse
 anthropic
-screeninfo
-sentence_transformers
-nltk
-thefuzz
 beautifulsoup4
 google-generativeai
 google-genai
 duckduckgo-search
-pypdf
-PyMuPDF
-opencv-python
-librosa
 openai
 jinja2
-pandas
-matplotlib
-IPython
-ollama
 requests
 markdown
 PyYAML
 langchain
 langchain_community
-pyautogui
 pygments
-pyttsx3
-kuzu
-chromadb
 termcolor
 colorama
 python-dotenv
@@ -36,4 +20,19 @@ pytest
 googlesearch-python
 flask
 flask_cors
+librosa
+pandas
+matplotlib
+IPython
+pyautogui
+nltk
+thefuzz
+pypdf
+PyMuPDF
+screeninfo
+sentence_transformers
+opencv-python
+ollama
+kuzu
+chromadb
 diffusers

{npcsh-0.3.26 → npcsh-0.3.27}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 import os
 import site
-
+import sys
 import platform
 from pathlib import Path
 
@@ -53,57 +53,79 @@ To configure your API keys and preferences.
     return ""  # Return empty string for non-Windows platforms
 
 
+# Define core (lite) requirements
+core_requirements = [
+    "redis",
+    "flask_sse",
+    "anthropic",
+    "beautifulsoup4",
+    "google-generativeai",
+    "google-genai",
+    "duckduckgo-search",
+    "openai",
+    "jinja2",
+    "requests",
+    "markdown",
+    "PyYAML",
+    "langchain",
+    "langchain_community",
+    "pygments",
+    "termcolor",
+    "colorama",
+    "python-dotenv",
+    "pytest",
+    "googlesearch-python",
+    "flask",
+    "flask_cors",
+    "librosa",
+    "pandas",
+    "matplotlib",
+    "IPython",
+    "pyautogui",
+    "nltk",
+    "thefuzz",
+    "pypdf",
+    "PyMuPDF",
+    "screeninfo",
+]
+
+# Define full requirements
+full_requirements = core_requirements + [
+    "sentence_transformers",
+    "opencv-python",
+    "ollama",
+    "kuzu",
+    "chromadb",
+    "diffusers",
+]
+
+# Define audio requirements
+audio_requirements = [
+    "openai-whisper",
+    "pyaudio",
+    "gtts",
+    "playsound==1.2.2",
+    "pyttsx3",
+]
+
 extra_files = package_files("npcsh/npc_team/")
 
+# Choose requirements based on environment variables
+is_lite = os.environ.get("NPCSH_LITE_INSTALL", "").lower() == "true"
+is_audio = os.environ.get("NPCSH_AUDIO_INSTALL", "").lower() == "true"
+
+if is_lite:
+    requirements = core_requirements
+elif is_audio:
+    requirements = full_requirements + audio_requirements
+else:
+    requirements = full_requirements
+
 setup(
     name="npcsh",
-    version="0.3.26",
+    version="0.3.27",
     packages=find_packages(exclude=["tests*"]),
-    install_requires=[
-        "redis",
-        "flask_sse",
-        "anthropic",
-        "screeninfo",
-        "sentence_transformers",
-        "nltk",
-        "thefuzz",
-        "beautifulsoup4",
-        "google-generativeai",
-        "google-genai",
-        "duckduckgo-search",
-        "pypdf",
-        "PyMuPDF",
-        "opencv-python",
-        "librosa",
-        "openai",
-        "jinja2",
-        "pandas",
-        "matplotlib",
-        "IPython",
-        "ollama",
-        "requests",
-        "markdown",
-        "PyYAML",
-        "langchain",
-        "langchain_community",
-        "pyautogui",
-        # "openai-whisper",
-        # "pyaudio",
-        # "gtts",
-        # "playsound==1.2.2",
-        "pygments",
-        "pyttsx3",
-        "kuzu",
-        "chromadb",
-        "termcolor",
-        "colorama",
-        "python-dotenv",
-        "pytest",
-        "googlesearch-python",
-        "flask",
-        "flask_cors",
-        "diffusers",
-    ],
+    install_requires=requirements,
     entry_points={
         "console_scripts": [
             "npcsh=npcsh.shell:main",
@@ -124,6 +146,7 @@ setup(
     data_files=[("npcsh/npc_team", extra_files)],
     python_requires=">=3.10",
 )
+
 # Print setup message only on Windows
 if platform.system() == "Windows":
     print(get_setup_message())