integrating agentic chroma

src/ai_generators/ollama_md_generator.py

@@ -126,7 +126,6 @@ class OllamaGenerator:
                 {self.content}
             """
         try:
-<<<<<<< HEAD
             query_embed = self.ollama_client.embed(model=self.embed_model, input=prompt_system)['embeddings']
             collection = self.load_to_vector_db()
             collection_query = collection.query(query_embeddings=query_embed, n_results=100)
@@ -147,24 +146,6 @@ class OllamaGenerator:
             #print ("Markdown Generated")
             #print (self.response)
             return self.response#['message']['content']
-=======
-            self.response = self.ollama_client.chat(model=self.ollama_model,
-                                                    messages=[
-                    {
-                        'role': 'user',
-                        'content': f'{prompt}',
-                    },
-                ])
-
-            # the deepseek model returns <think> this removes those tabs from the output
-<<<<<<< HEAD
-            return re.sub(r"<think|.\n\r+?|([^;]*)\/think>",'',self.response['message']['content'])
->>>>>>> e1a24af (get rid of think tags)
-=======
-            # return re.sub(r"<think|.\n\r+?|([^;]*)\/think>",'',self.response['message']['content'])
-            return self.response['message']['content']
-
->>>>>>> d45f0be (env set up for remote)
 
         except Exception as e:
             raise Exception(f"Failed to generate markdown: {e}")
@@ -173,12 +154,9 @@ class OllamaGenerator:
         with open(filename, "w") as f:
             f.write(self.generate_markdown())
 
-<<<<<<< HEAD
     def generate_commit_message(self):
         prompt_system = "You are a blog creator commiting a piece of content to a central git repo"
         prompt_human = f"Generate a 5 word git commit message describing {self.response}"
         messages = [("system", prompt_system), ("human", prompt_human),]
         commit_message = self.llm.invoke(messages).text()
         return commit_message
-=======
->>>>>>> e1a24af (get rid of think tags)

src/main.py

@@ -19,15 +19,8 @@ for note in tril_notes:
     print(tril_notes[note]['title'])
     # print(tril_notes[note]['content'])
     print("Generating Document")
-<<<<<<< HEAD
 
-=======
-    ai_gen = omg.OllamaGenerator(tril_notes[note]['title'],
-                                 tril_notes[note]['content'],
-                                 "openthinker:7b")
->>>>>>> d35a456 (set up chroma)
     os_friendly_title = convert_to_lowercase_with_underscores(tril_notes[note]['title'])
-<<<<<<< HEAD
     ai_gen = omg.OllamaGenerator(os_friendly_title,
                                  tril_notes[note]['content'],
                                  tril_notes[note]['title'])
@@ -39,6 +32,3 @@ for note in tril_notes:
     git_pass = os.environ["GIT_PASS"]
     repo_manager = git_repo.GitRepository("blog/", git_user, git_pass)
     repo_manager.create_copy_commit_push(blog_path, os_friendly_title, commit_message)
-=======
-    ai_gen.save_to_file(f"./generated_files/{os_friendly_title}.md")
->>>>>>> d45f0be (env set up for remote)