kssrag 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

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.dist-info → kssrag-0.1.1.dist-info}/METADATA

@@ -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.dist-info → kssrag-0.1.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 kssrag/__init__.py,sha256=N1XfR8IRKtEJAzcOVyHnKXtgx-ztlrSFtFwiVkGbAX8,2041
-kssrag/cli.py,sha256=eCJ8VhCzWIYUQEIPWJnzPPE298GWxWS8ienwf2eXpgk,6461
+kssrag/cli.py,sha256=r_W5M_sxq40jKpydmVOAzfyp9XDT_yHT4TRukPhRMhk,7300
 kssrag/config.py,sha256=vJUDJVfCoBy5QXw2YBGJBE6y_e5du4fq65EhT2pz3f0,6101
 kssrag/kssrag.py,sha256=vy3oCHeHFAp_dJW0JjLbTxeEwCcwtXuOL_Ejmv0qz8Y,5251
 kssrag/server.py,sha256=HR60UqVlimo9UrExv_zTh6faE4ueVuz-QKK7UtZxmAA,4029
@@ -15,12 +15,11 @@ kssrag/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kssrag/utils/document_loaders.py,sha256=TLfi1-4YIyvuyIqiNBqLG9zl_FoT65s-zp37yqYSq38,1388
 kssrag/utils/helpers.py,sha256=ejgViGTixsjPt0U26xZKodkqlB2wRMQlDzCmCjskYfg,1899
 kssrag/utils/preprocessors.py,sha256=_kbeZOWnbqbKKSBiyRP8QZAKx9uYMXgHfARcWBqC3JU,938
-kssrag-0.1.0.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_basic.py,sha256=JdBBRpP9wOo4BvvZTisidP40gGyK_azUoewJpoJaa5M,1275
 tests/test_vectorstores.py,sha256=YOwI2bfqprzbq8ahIw4pbbbEOaKGcg-XPcLCO7WiLxE,1474
-kssrag-0.1.0.dist-info/METADATA,sha256=L1eP9IPzxipFkJbDn_AAXbdEWpsCQfFiO0ilkK7wScQ,10342
-kssrag-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kssrag-0.1.0.dist-info/entry_points.txt,sha256=g4tQj5YUqPK3Osb9BI85tsErxleSBUENiqlnX0fWK5M,43
-kssrag-0.1.0.dist-info/top_level.txt,sha256=sO9LGINa0GEjLoHTtufpz01yM5SmeTw6M4zWHEF0R2s,13
-kssrag-0.1.0.dist-info/RECORD,,
+kssrag-0.1.1.dist-info/METADATA,sha256=cXlOZ6nE5WeL9eP0puBRyq5S-5pf4R5q5iBhdoHG15I,10592
+kssrag-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kssrag-0.1.1.dist-info/entry_points.txt,sha256=g4tQj5YUqPK3Osb9BI85tsErxleSBUENiqlnX0fWK5M,43
+kssrag-0.1.1.dist-info/top_level.txt,sha256=sO9LGINa0GEjLoHTtufpz01yM5SmeTw6M4zWHEF0R2s,13
+kssrag-0.1.1.dist-info/RECORD,,