starred/minne
1
0
Fork 0
mirror of https://github.com/perstarkse/minne.git synced 2026-03-30 06:12:00 +02:00
Code Issues Packages Projects Releases 10 Wiki Activity

chore: cleaning & deref

Browse Source
This commit is contained in:
Per Stark
2024-11-07 08:30:55 +01:00
parent 3f06bf969a
commit 0f491f88d3
4 changed files with 41 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/utils/llm.rs
View File

@@ -1,12 +1,10 @@
use crate::models::graph_entities::GraphMapper;
use crate::models::graph_entities::KnowledgeEntity;
use crate::models::graph_entities::KnowledgeEntityType;
use crate::models::graph_entities::KnowledgeRelationship;
use crate::models::graph_entities::{GraphMapper, KnowledgeEntity, KnowledgeEntityType, KnowledgeRelationship};
use crate::models::text_content::ProcessingError;
use crate::surrealdb::SurrealDbClient;
use async_openai::types::{CreateChatCompletionRequestArgs, ChatCompletionRequestUserMessage, ChatCompletionRequestSystemMessage };
use serde::{Deserialize, Serialize};
use serde_json::json;
use surrealdb::engine::remote::ws::Client;
use surrealdb::Surreal;
use tracing::{info,debug};
use uuid::Uuid;
@@ -86,14 +84,20 @@ pub async fn create_json_ld(
category: &str,
instructions: &str,
text: &str,
db_client: &SurrealDbClient,
db_client: &Surreal<Client>,
) -> Result<LLMGraphAnalysisResult, ProcessingError> {
// Get the nodes from the database
let entities: Vec<KnowledgeEntity> = db_client.client.select("knowledge_entity").await?;
let entities: Vec<KnowledgeEntity> = db_client.select("knowledge_entity").await?;
info!("{:?}", entities);
let deleted: Vec<KnowledgeEntity> = db_client.client.delete("knowledge_entity").await?;
info! {"{:?}", deleted};
let deleted: Vec<KnowledgeEntity> = db_client.delete("knowledge_entity").await?;
info! {"{:?} KnowledgeEntities deleted", deleted.len()};
let relationships: Vec<KnowledgeRelationship> = db_client.select("knowledge_relationship").await?;
info!("{:?}", relationships);
let relationships_deleted: Vec<KnowledgeRelationship> = db_client.delete("knowledge_relationship").await?;
info!("{:?} Relationships deleted", relationships_deleted.len());
let client = async_openai::Client::new();
let schema = json!({