pr fixes

Author: ea-rus

mindsdb_sdk/jobs.py

@@ -38,7 +38,10 @@ def __repr__(self):
 
     def __enter__(self):
         if self._create_callback is None:
-            raise RuntimeError('It can not be used to create context')
+            raise ValueError("The job is already created and can't be used to create context."
+                               " To be able to use context: create job without 'query_str' parameter: "
+                               "\n>>> with con.jobs.create('j1') as job:"
+                               "\n>>>    job.add_query(...)")
         set_saving(f'job-{self.name}')
 
         return self
@@ -47,7 +50,7 @@ def __exit__(self, type, value, traceback):
         set_saving(None)
         if type is None:
             if len(self._queries) == 0:
-                raise RuntimeError('No queries were added to job')
+                raise ValueError('No queries were added to job')
 
             query_str = '; '.join(self._queries)
 
@@ -79,7 +82,7 @@ def add_query(self, query: Union[Query, str]):
 
             query = query.sql
         elif not isinstance(query, str):
-            raise RuntimeError(f'Unable to use add this object as a query: {query}. Try to use sql string instead')
+            raise ValueError(f'Unable to use add this object as a query: {query}. Try to use sql string instead')
         self._queries.append(query)
 
     def get_history(self) -> pd.DataFrame:
@@ -164,7 +167,7 @@ def create(
         Usage options:
 
         Option 1: to use string query
-        All job tasks could be passed as string with sql queries. Job is created emmideiately
+        All job tasks could be passed as string with sql queries. Job is created immediately
 
         >>> job = con.jobs.create('j1', query_str='retrain m1; show models', repeat_min=1):
 

mindsdb_sdk/utils/context.py

@@ -3,23 +3,44 @@
 context_storage = ContextVar('create_context')
 
 
-def set_context(name, value):
+def set_context(name: str, value: str):
+    """
+    Set context value to variable
+
+    :param name: variable name
+    :param value: variable value
+    """
     data = context_storage.get({})
     data[name] = value
 
     context_storage.set(data)
 
 
-def get_context(name):
+def get_context(name: str) -> str:
+    """
+    Get context value fom variable
+
+    :param name: variable name
+    :return: variable value
+    """
 
     data = context_storage.get({})
     return data.get(name)
 
 
-def set_saving(name):
+def set_saving(name: str):
+    """
+    Set name of saving object to context
+
+    :param name: namve of the object
+    """
     set_context('saving', name)
 
 
-def is_saving():
+def is_saving() -> bool:
+    """
+    Returns true if object is saved at the moment
+    """
+
     return get_context('saving') is not None