kssrag 0.1.0__tar.gz → 0.1.1__tar.gz

{kssrag-0.1.0 → kssrag-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kssrag
-Version: 0.1.0
+Version: 0.1.1
 Summary: A flexible Retrieval-Augmented Generation framework by Ksschkw
 Home-page: https://github.com/Ksschkw/kssrag
 Author: Ksschkw
@@ -22,7 +22,6 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.8, <4
 Description-Content-Type: text/markdown
-License-File: LICENSE
 Requires-Dist: fastapi>=0.104.0
 Requires-Dist: uvicorn>=0.24.0
 Requires-Dist: python-dotenv>=1.0.0
@@ -55,7 +54,6 @@ Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: keywords
-Dynamic: license-file
 Dynamic: project-url
 Dynamic: provides-extra
 Dynamic: requires-dist
@@ -65,6 +63,8 @@ Dynamic: summary
 # 🚀 KSS RAG - Knowledge Retrieval Augmented Generation Framework
 
 > Built by [Ksschkw](https://github.com/Ksschkw)
+\
+> View at: [PyPi](https://pypi.org/project/kssrag/0.1.0/)
 
 ![Python Version](https://img.shields.io/badge/python-3.8%2B-blue)
 ![License](https://img.shields.io/badge/license-MIT-green)
@@ -118,10 +118,10 @@ print(response)
 export OPENROUTER_API_KEY="your_key_here"
 
 # Query documents
-python -m kssrag.cli query --file document.txt --query "Main ideas?"
+python -m kssrag.cli query --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --query "Main ideas?"
 
 # Start API server
-python -m kssrag.cli server --file document.txt --port 8000
+python -m kssrag.cli server --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --port 8000
 ```
 
 ## 🐳 Docker Deployment
@@ -265,7 +265,7 @@ python -m pytest --cov=kssrag tests/
 **CLI Command Not Found**
 ```bash
 # Use module syntax on Windows
-python -m kssrag.cli query --file document.txt --query "Your question"
+python -m kssrag.cli query --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --query "Your question"
 ```
 
 **FAISS Windows Issues**

{kssrag-0.1.0 → kssrag-0.1.1}/README.md

@@ -1,6 +1,8 @@
 # 🚀 KSS RAG - Knowledge Retrieval Augmented Generation Framework
 
 > Built by [Ksschkw](https://github.com/Ksschkw)
+\
+> View at: [PyPi](https://pypi.org/project/kssrag/0.1.0/)
 
 ![Python Version](https://img.shields.io/badge/python-3.8%2B-blue)
 ![License](https://img.shields.io/badge/license-MIT-green)
@@ -54,10 +56,10 @@ print(response)
 export OPENROUTER_API_KEY="your_key_here"
 
 # Query documents
-python -m kssrag.cli query --file document.txt --query "Main ideas?"
+python -m kssrag.cli query --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --query "Main ideas?"
 
 # Start API server
-python -m kssrag.cli server --file document.txt --port 8000
+python -m kssrag.cli server --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --port 8000
 ```
 
 ## 🐳 Docker Deployment
@@ -201,7 +203,7 @@ python -m pytest --cov=kssrag tests/
 **CLI Command Not Found**
 ```bash
 # Use module syntax on Windows
-python -m kssrag.cli query --file document.txt --query "Your question"
+python -m kssrag.cli query --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --query "Your question"
 ```
 
 **FAISS Windows Issues**

{kssrag-0.1.0 → kssrag-0.1.1}/kssrag/cli.py

@@ -1,5 +1,6 @@
 import argparse
 import sys
+import os  # Add this import if not already present
 from .utils.document_loaders import load_document, load_json_documents
 from .core.chunkers import TextChunker, JSONChunker, PDFChunker
 from .core.vectorstores import BM25VectorStore, FAISSVectorStore, TFIDFVectorStore, HybridVectorStore, HybridOfflineVectorStore
@@ -24,6 +25,7 @@ def main():
                              choices=["bm25", "faiss", "tfidf", "hybrid_online", "hybrid_offline"], 
                              help="Vector store type")
     query_parser.add_argument("--top-k", type=int, default=config.TOP_K, help="Number of results to retrieve")
+    query_parser.add_argument("--system-prompt", type=str, help="Path to a file containing the system prompt or the prompt text itself")
     
     # Server command
     server_parser = subparsers.add_parser("server", help="Start the RAG API server")
@@ -35,12 +37,22 @@ def main():
                               help="Vector store type")
     server_parser.add_argument("--port", type=int, default=config.SERVER_PORT, help="Port to run server on")
     server_parser.add_argument("--host", type=str, default=config.SERVER_HOST, help="Host to run server on")
+    server_parser.add_argument("--system-prompt", type=str, help="Path to a file containing the system prompt or the prompt text itself")
     
     args = parser.parse_args()
     
     # Validate config
     validate_config()
     
+    def load_system_prompt(prompt_arg):
+        """Load system prompt from file or use as text"""
+        if not prompt_arg:
+            return None
+        if os.path.exists(prompt_arg):
+            with open(prompt_arg, 'r', encoding='utf-8') as f:
+                return f.read()
+        return prompt_arg
+    
     if args.command == "query":
         # Load and process document
         if args.format == "text":
@@ -78,7 +90,8 @@ def main():
         # Create retriever and agent
         retriever = SimpleRetriever(vector_store)
         llm = OpenRouterLLM()
-        agent = RAGAgent(retriever, llm)
+        system_prompt = load_system_prompt(args.system_prompt)
+        agent = RAGAgent(retriever, llm, system_prompt=system_prompt)
         
         # Query and print response
         response = agent.query(args.query, top_k=args.top_k)
@@ -122,7 +135,8 @@ def main():
         # Create retriever and agent
         retriever = SimpleRetriever(vector_store)
         llm = OpenRouterLLM()
-        agent = RAGAgent(retriever, llm)
+        system_prompt = load_system_prompt(args.system_prompt)
+        agent = RAGAgent(retriever, llm, system_prompt=system_prompt)
         
         # Create and run server
         from .server import create_app

{kssrag-0.1.0 → kssrag-0.1.1}/kssrag.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kssrag
-Version: 0.1.0
+Version: 0.1.1
 Summary: A flexible Retrieval-Augmented Generation framework by Ksschkw
 Home-page: https://github.com/Ksschkw/kssrag
 Author: Ksschkw
@@ -22,7 +22,6 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.8, <4
 Description-Content-Type: text/markdown
-License-File: LICENSE
 Requires-Dist: fastapi>=0.104.0
 Requires-Dist: uvicorn>=0.24.0
 Requires-Dist: python-dotenv>=1.0.0
@@ -55,7 +54,6 @@ Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: keywords
-Dynamic: license-file
 Dynamic: project-url
 Dynamic: provides-extra
 Dynamic: requires-dist
@@ -65,6 +63,8 @@ Dynamic: summary
 # 🚀 KSS RAG - Knowledge Retrieval Augmented Generation Framework
 
 > Built by [Ksschkw](https://github.com/Ksschkw)
+\
+> View at: [PyPi](https://pypi.org/project/kssrag/0.1.0/)
 
 ![Python Version](https://img.shields.io/badge/python-3.8%2B-blue)
 ![License](https://img.shields.io/badge/license-MIT-green)
@@ -118,10 +118,10 @@ print(response)
 export OPENROUTER_API_KEY="your_key_here"
 
 # Query documents
-python -m kssrag.cli query --file document.txt --query "Main ideas?"
+python -m kssrag.cli query --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --query "Main ideas?"
 
 # Start API server
-python -m kssrag.cli server --file document.txt --port 8000
+python -m kssrag.cli server --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --port 8000
 ```
 
 ## 🐳 Docker Deployment
@@ -265,7 +265,7 @@ python -m pytest --cov=kssrag tests/
 **CLI Command Not Found**
 ```bash
 # Use module syntax on Windows
-python -m kssrag.cli query --file document.txt --query "Your question"
+python -m kssrag.cli query --file document.txt --system-prompt custom_prompt.txt(`or just insert plain text here in quotes`) --query "Your question"
 ```
 
 **FAISS Windows Issues**

{kssrag-0.1.0 → kssrag-0.1.1}/kssrag.egg-info/SOURCES.txt

@@ -1,4 +1,3 @@
-LICENSE
 README.md
 setup.py
 ./kssrag/__init__.py

{kssrag-0.1.0 → kssrag-0.1.1}/setup.py

@@ -6,7 +6,7 @@ long_description = (here / "README.md").read_text(encoding="utf-8")
 
 setup(
     name="kssrag",
-    version="0.1.0",
+    version="0.1.1",
     description="A flexible Retrieval-Augmented Generation framework by Ksschkw",
     long_description=long_description,
     long_description_content_type="text/markdown",