worqhat 0.0.1__py3-none-any.whl

worqhat/ai_models/text_gen.py

@@ -0,0 +1,177 @@
+import requests 
+import json
+import os
+def get_ai_response_v2(question="", 
+                       preserve_history=False, 
+                       randomness=0.5, 
+                       stream_data=False, 
+                       conversation_history=[], 
+                       training_data="", 
+                       response_type="text",
+                       api_key=None):
+    url = "https://api.worqhat.com/api/ai/content/v2"
+
+    payload = {
+        "question": question,
+        "preserve_history": preserve_history,
+        "randomness": randomness,
+        "stream_data": stream_data,
+        "conversation_history": conversation_history,
+        "training_data": training_data,
+        "response_type": response_type
+    }
+
+    headers = {"Content-Type": "application/json"}
+
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+
+    if api_key:
+        headers["Authorization"] = "Bearer " + api_key
+    else:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    response = requests.post(url, json=payload, headers=headers, stream=stream_data)
+
+    if stream_data:
+        for line in response.iter_lines():
+            if line:
+                if line.startswith(b'data:'):
+                    json_content = line[len(b'data:'):].decode('utf-8').strip()
+                    print(json_content)
+    else:
+        json_response = response.text
+        try:
+            json_data = json.loads(json_response)
+            return(json.dumps(json_data, indent=4))
+        except json.JSONDecodeError:
+            print(json_response)
+
+
+def get_ai_response_v3(question="", 
+                       preserve_history=False, 
+                       randomness=0.5, 
+                       stream_data=False, 
+                       conversation_history=[], 
+                       training_data="", 
+                       response_type="text",
+                       api_key=None):
+    url = "https://api.worqhat.com/api/ai/content/v3"
+
+    payload = {
+        "question": question,
+        "preserve_history": preserve_history,
+        "randomness": randomness,
+        "stream_data": stream_data,
+        "conversation_history": conversation_history,
+        "training_data": training_data,
+        "response_type": response_type
+    }
+
+    headers = {"Content-Type": "application/json"}
+
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+
+    if api_key:
+        headers["Authorization"] = "Bearer " + api_key
+    else:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    response = requests.post(url, json=payload, headers=headers, stream=stream_data)
+
+    if stream_data:
+        for line in response.iter_lines():
+            if line:
+                if line.startswith(b'data:'):
+                    json_content = line[len(b'data:'):].decode('utf-8').strip()
+                    print(json_content)
+    else:
+        json_response = response.text
+        try:
+            json_data = json.loads(json_response)
+            print(json.dumps(json_data, indent=4))
+        except json.JSONDecodeError:
+            print(json_response)
+
+
+def get_alpha_ai_response(question="", 
+                          preserve_history=False, 
+                          randomness=0.5, 
+                          stream_data=False, 
+                          conversation_history=[], 
+                          training_data="", 
+                          response_type="text",
+                          api_key=None):
+    url = "https://api.worqhat.com/api/ai/content/v3/alpha"
+    payload = {
+        "question": question,
+        "preserve_history": preserve_history,
+        "randomness": randomness,
+        "stream_data": stream_data,
+        "conversation_history": conversation_history,
+        "training_data": training_data,
+        "response_type": response_type
+    }
+    headers = {"Content-Type": "application/json"}
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if api_key:
+        headers["Authorization"] = "Bearer " + api_key
+    else:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    response = requests.request("POST",url, json=payload, headers=headers)
+    if stream_data:
+        for line in response.iter_lines():
+            if line:
+                if line.startswith(b'data:'):
+                    json_content = line[len(b'data:'):].decode('utf-8').strip()
+                    print(json_content)
+    else:
+        json_response = response.text
+        try:
+            json_data = json.loads(json_response)
+            print(json.dumps(json_data, indent=4))
+        except json.JSONDecodeError:
+            print(json_response)
+
+
+def get_large_ai_response_v2(question="", 
+                             dataset_id="", 
+                             preserve_history=True, 
+                             randomness=0.5, 
+                             stream_data=False, 
+                             conversation_history=[], 
+                             instructions=None,
+                             api_key=None):
+    url = "https://api.worqhat.com/api/ai/content/v2-large/answering"
+    payload = {
+        "question": question,
+        "datasetId": dataset_id,
+        "preserve_history": preserve_history,
+        "randomness": randomness,
+        "stream_data": stream_data,
+        "conversation_history": conversation_history,
+        "instructions": instructions
+    }
+    headers = {"Content-Type": "application/json"}
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if api_key:
+        headers["Authorization"] = "Bearer " + api_key
+    else:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    response = requests.request("POST",url, json=payload, headers=headers)
+    if stream_data:
+        for line in response.iter_lines():
+            if line:
+                if line.startswith(b'data:'):
+                    json_content = line[len(b'data:'):].decode('utf-8').strip()
+                    print(json_content)
+    else:
+        json_response = response.text
+        try:
+            json_data = json.loads(json_response)
+            print(json.dumps(json_data, indent=4))
+        except json.JSONDecodeError:
+            print(json_response)

worqhat/database_management/Edit.py

@@ -0,0 +1,140 @@
+import requests
+import os 
+
+
+def add_data_to_collection(collection='', doc_id='', data='',api_key=None ):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/add"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "collection": collection,
+        "docId": doc_id,
+        "data": data
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+
+def update_data_in_collection( collection='', doc_id='', data='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/update"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "collection": collection,
+        "docId": doc_id,
+        "data": data
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+
+    return response.text
+
+def increment_field_in_collection(collection='', doc_id='', field='', increment_value='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/increment"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "collection": collection,
+        "docId": doc_id,
+        "field": field,
+        "increment": increment_value
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+
+    return response.text
+
+def update_array_in_collection(collection='', doc_id='', field='', value='',api_key=None ):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/array/update/add"
+    payload = {
+        "collection": collection,
+        "docId": doc_id,
+        "field": field,
+        "arrayUnion": value
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+
+    return response.text
+
+def remove_array_element_from_collection(collection='', doc_id='', field='', value='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/array/update/remove"
+    payload = {
+        "collection": collection,
+        "docId": doc_id,
+        "field": field,
+        "arrayRemove": value
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+
+    return response.text
+
+def delete_doc_from_collection(collection='', doc_id='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/delete"
+    payload = {
+        "collection": collection,
+        "docId": doc_id
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+
+    return response.text

worqhat/database_management/Read.py

@@ -0,0 +1,152 @@
+import requests
+import os 
+
+def fetch_all_collections(api_key=None):
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/fetch-all"
+    headers = {
+        "Authorization": "Bearer " + api_key
+    }
+
+    response = requests.post("POST",url, headers=headers)
+
+    return response.text
+
+def fetch_all_docs_from_collection(collection='', output_type="json",api_key=None,):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/fetch/all"
+    payload = {
+        "collection": collection,
+        "outputType": output_type
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+
+def fetch_doc_from_collection( collection='', doc_id='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/fetch/document"
+    payload = {
+        "collection": collection,
+        "documentId": doc_id
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+
+def fetch_doc_count_by_field(collection='', field='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/fetch/count"
+    payload = {
+        "collection": collection,
+        "key": field
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+
+def fetch_unique_keys_ordered(collection='', field='', order_by='', order_type='',api_key=None):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/fetch/unique"
+    payload = {
+        "collection": collection,
+        "key": field,
+        "orderBy": order_by,
+        "orderType": order_type
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+
+def fetch_docs_by_query(collection='', queries='', compounding="and", order_by=None, order_type="asc", limit=None, start_after=None, output_type="json",api_key=None ):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/fetch/query"
+    payload = {
+        "collection": collection,
+        "queries": queries,
+        "compounding": compounding,
+        "orderBy": order_by,
+        "orderType": order_type,
+        "limit": limit,
+        "startAfter": start_after,
+        "outputType": output_type
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+
+def fetch_docs_by_natural_query(collection='', query='', output_type="json",api_key=None ):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/data/fetch/natural-query"
+    payload = {
+        "collection": collection,
+        "query": query,
+        "outputType": output_type
+    }
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text
+

worqhat/database_management/__init__.py

@@ -0,0 +1,5 @@
+__all__ = ['collection', 'Edit', 'Read']
+
+from .collection import *
+from .Edit import *
+from .Read import *

worqhat/database_management/collection.py

@@ -0,0 +1,25 @@
+import requests
+import os 
+
+def create_collection(collection='', collection_schema='', collection_sort_by='',api_key=None ):
+    if collection=='':
+        return ("Please enter a Collection name")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/collections/create"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "collection": collection,
+        "collectionSchema": collection_schema,
+        "collectionSortBy": collection_sort_by
+    }
+
+    response = requests.post("POST",url, json=payload, headers=headers)
+
+    return response.text

worqhat/test/test_ai_search.py

@@ -0,0 +1,38 @@
+import unittest
+from unittest.mock import MagicMock
+from ..src.ai_models.ai_search import search_ai_v2, search_ai_v3
+
+class TestAISearch(unittest.TestCase):
+
+    def setUp(self):
+        # Mock API key for testing
+        self.api_key = "mock_api_key"
+
+    def test_search_ai_v2(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test result for AI v2"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('ai_search.requests', requests_mock):
+            result = search_ai_v2(api_key=self.api_key, question="Test question")
+
+        self.assertEqual(result, '{"result": "Test result for AI v2"}')
+        requests_mock.request.assert_called_once()
+
+    def test_search_ai_v3(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test result for AI v3"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('ai_search.requests', requests_mock):
+            result = search_ai_v3(api_key=self.api_key, question="Test question")
+
+        self.assertEqual(result, '{"result": "Test result for AI v3"}')
+        requests_mock.request.assert_called_once()
+
+if __name__ == '__main__':
+    unittest.main()

worqhat/test/test_content_mod.py

@@ -0,0 +1,39 @@
+import unittest
+from unittest.mock import MagicMock
+import requests
+from ..src.ai_models.content_mod import content_moderation,image_moderation
+
+class TestModeration(unittest.TestCase):
+
+    def setUp(self):
+        # Mock API key for testing
+        self.api_key = "mock_api_key"
+
+    def test_content_moderation(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test content moderation result"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('moderation.requests', requests_mock):
+            result = content_moderation(api_key=self.api_key, text_content="Test text content")
+
+        self.assertEqual(result, '{"result": "Test content moderation result"}')
+        requests_mock.request.assert_called_once()
+
+    def test_image_moderation(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test image moderation result"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('moderation.requests', requests_mock):
+            result = image_moderation(api_key=self.api_key, image_file=MagicMock(name="mock_image_file"))
+
+        self.assertEqual(result, '{"result": "Test image moderation result"}')
+        requests_mock.request.assert_called_once()
+
+if __name__ == '__main__':
+    unittest.main()

worqhat/test/test_image_analysis.py

@@ -0,0 +1,52 @@
+import unittest
+from unittest.mock import MagicMock
+import requests
+from ..src.ai_models.image_analysis import image_analysis, face_detection, facial_comparison
+
+class TestImageFunctions(unittest.TestCase):
+
+    def setUp(self):
+        # Mock API key for testing
+        self.api_key = "mock_api_key"
+
+    def test_image_analysis(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test image analysis result"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('your_module.requests', requests_mock):
+            result = image_analysis(api_key=self.api_key, images=[MagicMock(name="mock_image")], question="Test question")
+
+        self.assertEqual(result, '{"result": "Test image analysis result"}')
+        requests_mock.request.assert_called_once()
+
+    def test_face_detection(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test face detection result"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('your_module.requests', requests_mock):
+            result = face_detection(api_key=self.api_key, image_file=MagicMock(name="mock_image_file"))
+
+        self.assertEqual(result, '{"result": "Test face detection result"}')
+        requests_mock.request.assert_called_once()
+
+    def test_facial_comparison(self):
+        # Mock response
+        mock_response = MagicMock()
+        mock_response.text = '{"result": "Test facial comparison result"}'
+        requests_mock = MagicMock()
+        requests_mock.request.return_value = mock_response
+
+        with unittest.mock.patch('your_module.requests', requests_mock):
+            result = facial_comparison(api_key=self.api_key, source_image_file=MagicMock(name="mock_source_image_file"), target_image_file=MagicMock(name="mock_target_image_file"))
+
+        self.assertEqual(result, '{"result": "Test facial comparison result"}')
+        requests_mock.request.assert_called_once()
+
+if __name__ == '__main__':
+    unittest.main()