Carlex22222 commited on
Commit
d072ddf
·
verified ·
1 Parent(s): 6681346

Update aduc_framework/types.py

Browse files
Files changed (1) hide show
  1. aduc_framework/types.py +20 -15
aduc_framework/types.py CHANGED
@@ -2,23 +2,32 @@
2
  #
3
  # Copyright (C) August 4, 2025 Carlos Rodrigues dos Santos
4
  #
5
- # Versão 4.0.0 (DNA with Input Provenance)
6
  #
7
- # Este arquivo define as estruturas de dados centrais para o Aduc Framework
8
- # usando Pydantic. Ele serve como a fonte única da verdade para a forma de
9
- # todos os dados que fluem através do sistema, incluindo o "DNA Digital"
10
- # completo de uma geração, que agora rastreia a proveniência de cada artefato.
11
 
12
  from pydantic import BaseModel, Field
13
  from typing import List, Dict, Any, Optional
 
 
14
 
15
- # --- Modelo de Proveniência (A "Receita") ---
 
 
 
 
 
 
 
 
 
16
 
17
  class LatentInput(BaseModel):
18
  """Representa uma única fonte de condicionamento usada para gerar um artefato."""
19
- caminho_origem: str = Field(description="Caminho para o arquivo de origem (.pt, .png) ou descrição da memória (ex: 'Memória Eco').")
20
- frame_alvo: int = Field(description="O frame no vídeo de destino onde este condicional é aplicado.")
21
- forca_condicionamento: float = Field(description="A força (peso) da influência deste condicional.")
22
 
23
  # --- Modelos de Parâmetros de Entrada ---
24
 
@@ -43,7 +52,7 @@ class GenerationParameters(BaseModel):
43
  """Agrega todos os parâmetros de configuração da geração."""
44
  pre_producao: Optional[PreProductionParams] = None
45
  producao: Optional[ProductionParams] = None
46
- pos_producao: Optional[Dict[str, Any]] = None # Para futuras expansões
47
 
48
  # --- Modelos de Artefatos Gerados (Componentes do DNA) ---
49
 
@@ -82,11 +91,7 @@ class VideoData(BaseModel):
82
  # --- O Modelo de Estado Principal (O DNA Completo) ---
83
 
84
  class GenerationState(BaseModel):
85
- """
86
- O "DNA Digital" completo de uma geração.
87
- Este é o objeto de estado central que flui através do framework,
88
- contendo todos os parâmetros, decisões e caminhos de artefatos.
89
- """
90
  parametros_geracao: GenerationParameters = Field(default_factory=GenerationParameters)
91
  Promt_geral: str = ""
92
  midias_referencia: List[MediaRef] = Field(default_factory=list)
 
2
  #
3
  # Copyright (C) August 4, 2025 Carlos Rodrigues dos Santos
4
  #
5
+ # Versão 4.1.0 (Corrected Internal Types)
6
  #
7
+ # Adiciona a definição do dataclass LatentConditioningItem, que é um tipo de
8
+ # dado interno essencial para a comunicação com o LTX Manager.
 
 
9
 
10
  from pydantic import BaseModel, Field
11
  from typing import List, Dict, Any, Optional
12
+ from dataclasses import dataclass
13
+ import torch
14
 
15
+ # --- Tipos de Dados Internos (Essencial para o Backend) ---
16
+
17
+ @dataclass
18
+ class LatentConditioningItem:
19
+ """Representa uma âncora de condicionamento no espaço latente para o LTX."""
20
+ latent_tensor: torch.Tensor
21
+ media_frame_number: int
22
+ conditioning_strength: float
23
+
24
+ # --- Modelo de Proveniência (A "Receita" do DNA) ---
25
 
26
  class LatentInput(BaseModel):
27
  """Representa uma única fonte de condicionamento usada para gerar um artefato."""
28
+ caminho_origem: str = Field(description="Caminho para a origem ou descrição da memória.")
29
+ frame_alvo: int = Field(description="O frame no vídeo de destino onde o condicional é aplicado.")
30
+ forca_condicionamento: float = Field(description="A força (peso) da influência do condicional.")
31
 
32
  # --- Modelos de Parâmetros de Entrada ---
33
 
 
52
  """Agrega todos os parâmetros de configuração da geração."""
53
  pre_producao: Optional[PreProductionParams] = None
54
  producao: Optional[ProductionParams] = None
55
+ pos_producao: Optional[Dict[str, Any]] = None
56
 
57
  # --- Modelos de Artefatos Gerados (Componentes do DNA) ---
58
 
 
91
  # --- O Modelo de Estado Principal (O DNA Completo) ---
92
 
93
  class GenerationState(BaseModel):
94
+ """O "DNA Digital" completo de uma geração."""
 
 
 
 
95
  parametros_geracao: GenerationParameters = Field(default_factory=GenerationParameters)
96
  Promt_geral: str = ""
97
  midias_referencia: List[MediaRef] = Field(default_factory=list)