unleash client

Reviewed-on: #16

requirements.txt

@@ -6,3 +6,4 @@ chromadb
 langchain-ollama
 PyJWT
 dotenv
+UnleashClient

src/ai_generators/ollama_md_generator.py

@@ -33,7 +33,7 @@ class OllamaGenerator:
             The title for the blog is {self.inner_title}.
             Do not output the title in the markdown.
             The basis for the content of the blog is:
-                {self.content}
+                <blog>{self.content}</blog>
             """
 
     def split_into_chunks(self, text, chunk_size=100):
@@ -127,10 +127,11 @@ class OllamaGenerator:
             you are more professional and will attempt to tone these down
             As this person produce the final version of this blog as a markdown document
             keeping in mind the context provided by the previous drafts.
+            You are to produce the content not placeholders for further editors
             The title for the blog is {self.inner_title}.
             Do not output the title in the markdown. Avoid repeated sentences
             The basis for the content of the blog is:
-                {self.content}
+                <blog>{self.content}</blog>
             """
         try:
             query_embed = self.ollama_client.embed(model=self.embed_model, input=prompt_system)['embeddings']
@@ -139,7 +140,7 @@ class OllamaGenerator:
             print("Showing pertinent info from drafts used in final edited edition")
             pertinent_draft_info = '\n\n'.join(collection.query(query_embeddings=query_embed, n_results=100)['documents'][0])
             #print(pertinent_draft_info)
-            prompt_human = f"""Generate the final, 3000 word, draft of the blog using this information from the drafts: {pertinent_draft_info}
+            prompt_human = f"""Generate the final, 3000 word, draft of the blog using this information from the drafts: <context>{pertinent_draft_info}</context>
                             - Only output in markdown, do not wrap in markdown tags, Only provide the draft not a commentary on the drafts in the context
                             """
             print("Generating final document")

unleash_client.py

@@ -0,0 +1,13 @@
+from UnleashClient import UnleashClient
+import asyncio
+
+client = UnleashClient(
+    url="http://192.168.178.160:30007/api/",
+    app_name="unleash-onboarding-python",
+    custom_headers={'Authorization': 'default:development.6uQIie4GdslTxgYAWVu35sRBjjBMPRRKw6vBj6mFsgFfvdXuy73GgLQg'}) # in production use environment variable
+
+client.initialize_client()
+
+while True:
+    print(client.is_enabled("crew_ai_integration"))
+    asyncio.run(asyncio.sleep(1))