Misc fixes/missing instructors.

airoboros/instructors/agent.py

@@ -0,0 +1,7 @@
+from airoboros.instructors.inline_qa import generate as generate_inline
+
+
+async def generate(instructor):
+    """Generator for agent/router training data."""
+    async for item in generate_inline(instructor, "agent", start_key="PROMPT"):
+        yield item

airoboros/instructors/cot.py

@@ -0,0 +1,7 @@
+from airoboros.instructors.inline_qa import generate as generate_inline
+
+
+async def generate(instructor):
+    """Generator for chain-of-thought training data."""
+    async for item in generate_inline(instructor, "cot"):
+        yield item

airoboros/self_instruct.py

@@ -319,7 +319,7 @@ async def _post(self, path: str, payload: Dict[str, Any]) -> Dict[str, Any]:
                 logger.debug(f"token usage: {self.used_tokens}")
                 return result
 
-    async def _post_no_exc(self, *a, **k) -> Dict[str, Any] | None:
+    async def _post_no_exc(self, *a, **k):
         """Post, ignoring all exceptions."""
         try:
             return await self._post(*a, **k)

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="airoboros",
-    version="2.0.0",
+    version="2.0.1",
     description="Updated and improved implementation of the self-instruct system.",
     long_description=long_description,
     long_description_content_type="text/markdown",